Sujong Lee

I'm an incoming Ph.D. student at NTU CCDS with Nanyang President's Graduate Scholarship (NPGS), advised by Prof. Atsushi Nitanda. Previously, I received B.Eng. in Electrical and Electronics Engineering with a Minor in Mathematics. I am also a researcher at Nanoforge AI. My research interests broadly lie in applied mathematics and their machine learning approach. I am specifically interested in probabilistic inference, including generative model and Monte Carlo methods.

Email  /  CV  /  Github  /  LinkedIn

profile photo

Research Interests

Generative Modeling: Investigating the theoretical foundations and efficiency of Diffusion Models, Flow Models, and their connection to Optimal Transport and Stochastic Control.

Neural Samplers: Developing advanced sampling techniques, particularly Monte Carlo Methods and their Neural Sampler approaches.

AI4Science: Applying machine learning to Materials Discovery, including research on the application of neural samplers to facilitate scientific innovation, especially on Materials.

Publications (* denote equal contribution)

Density of States-Intermediated Crystal Generation for Material Inverse Design Density of States-Intermediated Crystal Generation for Material Inverse Design
Sujong Lee*, J. Bae*, S. Shin*, H. Choi, C. Park
AI4Science@ICML, 2026
Neural Discrete Controlled Monte Carlo Samplers Neural Discrete Controlled Monte Carlo Samplers
Sujong Lee, P. Jutras-Dubé, B. Wen, R. Zhang
ProbML 2026

Experiences

NanoforgeAI AI Researcher Intern
NanoforgeAI, Korea
Oct 2025 - May 2026
Purdue University Research Intern
Purdue University (Remote)
Jun 2025 - Present
Supervisor: Ruqi Zhang
NTU Research Intern
ROSE Lab @ NTU, Singapore
Jan 2025 - May 2025
Supervisor: Bihan Wen
NTU URECA Research
NTU, Singapore
Sep 2021 - Jun 2022
Supervisor: Donguk Nam

Website template from Jon Barron.