Current Members

Jonathan Gregory
Jonathan Gregory
PhD Student (1st year)
Expected graduation: 2030
Trustworthy Machine Learning Privacy
Investigating privacy in machine learning techniques.
Abhinav Narayan Harish
Abhinav Narayan Harish
PhD Student (3rd year)
Expected graduation: 2028
Trustworthy Machine Learning Hallucination mitigation
Investigating robustness in machine learning techniques.
Shuibai Zhang
Swati Lavanya Govindaraju
PhD Student (2nd year)
Expected graduation: 2029
Generative models Diffusion models
Investigating diffusion models outside euclidean domains.
Shuibai Zhang
Jeffrey Wang
PhD Student (Incoming)
Expected graduation: 2030
Generative models Architecture design
Working on improving polynomial networks and architecture design.
Shuibai Zhang
Ioannis Chatzis
PhD Student (Incoming)
Expected graduation: 2031
Generative models Trustworthy ML
Giwon Shin
Giwon Shin
MS Student (2nd year)
BSc from Yonsei University (2023)
Expected graduation: 2026
Tensor decompositions Architecture design
Designing improved architectures for algorithmic tasks.
Shuibai Zhang
Shuibai Zhang
MS Student (2nd year)
Expected graduation: 2026
Generative models Error correction
Investigating error correction in diffusion language models.
Muhammad Ashiq
Samanyu Arora
Undergraduate Researcher
Senior, Computer Science
Expected graduation: 2027
Trustworthy Machine Learning Hallucination mitigation
Working on hallucination with guarantees.

Madison Alumni

Former group members who completed their degree at the University of Wisconsin–Madison. We are proud of their achievements and look forward to seeing their next steps:

UW–Madison Yiheng Zhang
Yiheng Zhang
BSc Computer Science  ·  2026
Senior Honors Thesis: Mixture Training of Diffusion Language Models for Math Reasoning Revision
Diffusion Language Models Trustworthy ML
First position since: CMU MS
UW–Madison Jeffrey
Jeffrey Wang
BSc Computer Science  ·  2026
Senior Honors Thesis:
Polynomial Networks
First position since: UW Madison PhD student
UW–Madison Muhammad
Muhammad Ashiq
BSc Computer Science  ·  2026
Senior Honors Thesis: A Theoretical Analysis of Mode Interpolation Hallucinations in ODE and SDE Diffusion Inference Procedures
Hallucination in diffusion models Trustworthy ML
First position since: Michigan PhD student
UW–Madison Radi
Radi Akbar
MSc Statistics  ·  2026
Architectures on tabular data
First position since: Scientist in American Family
UW–Madison Andrea
Andrea Tseng
BSc Computer Science  ·  2025
Attacks on Large Language Models Trustworthy ML
First position since: UW Madison PhD student

Alumni (outside of Madison)

I have had the privilege of collaborating with the following highly skilled students during their BSc/MSc/PhD degree:

Lars CPM Quaedvlieg (MSc 2025)

Meta -> Jump Trading

Yihang Chen (MSc 2024)

PhD student, UCLA

Justin Deschenaux (MSc 2024)

PhD student, EPFL

Pol Puigdemont (BSc 2024)

PhD student, EPFL

Henry Papadatos (MSc 2023)

Intern, UC Berkeley

Elias Abad Rocamora (BSc 2022, MSc 2023, PhD 2026)

Google DeepMind

Berk Mandiracioglu (MSc 2023)

UBS bank

Orest Gkini (MSc 2023)

Swisscom

Ke Wang (MSc 2023)

PhD student, EPFL

Alec Flowers (MSc 2023)

Nvidia

Yongtao Wu (MSc 2022)

PhD student, EPFL

Zeju Qiu (MSc 2022)

PhD student, Max-Planck-Institute for Intelligent Systems

Dimitrios Chalatsis (MSc 2022)

PhD student, Imperial College London

Zhenyu Zhu (MSc 2022)

PhD student, EPFL

Aleksandr Timofeev (MSc 2022)

Apple

Prateek Varshney (BSc 2021)

Research Associate at Google Research India

Moulik Choraria (MSc 2021)

PhD student, University of Illinois, Urbana-Champaign

Interested in Joining Our Group?

We are always looking for motivated students and researchers passionate about machine learning. If you're interested in pushing the boundaries of what's possible, we'd love to hear from you.

We welcome applications from PhD students, MSc students, undergraduate students, and visiting scholars. For undergraduate students, please make sure to check the following recomendations.

For PhD applicants, please apply through UW Madison's graduate school (CS or ECE).

Candidates with a foundation in mathematics and an analytical mindset are well aligned with my research priorities. Our recent work provides insights into the research directions we are pursuing. Although the volume of applications is high and I may not be able to respond to every inquiry individually, please be assured that every submission will be evaluated during our rigorous review process.

Contact Us