Hi, I’m Jingwen, a PhD student in the Columbia Core AI Lab (CAIL) and Theory of Computation Group at Columbia University, where I am extremely fortunate to be advised by Daniel Hsu and Alex Andoni. I obtained my undergraduate degree at UC San Diego, where I majored in mathematics and computer science. During my time at UCSD, I’m extremely honored to work with glorious Sanjoy Dasgupta and Russell Impagliazzo, who sparked my interest in ML theory and TCS research. I was also very lucky to work with Xiaolong Wang, who introduced me to CS research.
Research My recent research focuses on improving large language models with underlying scientific principles and understanding, spanning model architecture, training/optimization, and RL self-improvement.
Experience
-
Summer 2026Microsoft Research
-
Summer 2025University of Pennsylvania, Computer and Information ScienceHosted by Surbhi Goel
-
Fall 2024Simons Institute for the Theory of Computing, UC Berkeley
Publications
-
NeurIPS2026Fixed Universal TransformersAdvances in Neural Information Processing Systems (NeurIPS), 2026
-
ICML2026Less Data, Faster Training: repeating smaller datasets speeds up learning via sampling biasesInternational Conference on Machine Learning (ICML), 2026Contributed talk, Workshop on Scientific Methods for Understanding Deep Learning, ICLR 2026
-
ALT2026Group-realizable multi-group learning by minimizing empirical riskThe 37th International Conference on Algorithmic Learning Theory (ALT), 2026
-
NeurIPS2025Fast attention mechanisms: a tale of parallelismAdvances in Neural Information Processing Systems (NeurIPS), 2025
-
NeurIPS2024Group-wise oracle-efficient algorithms for online multi-group learningAdvances in Neural Information Processing Systems (NeurIPS), 2024
-
CVPR2022VideoINR: Learning Video Implicit Neural Representation for Continuous Space-Time Super-ResolutionConference on Computer Vision and Pattern Recognition (CVPR), 2022
Teaching and Service
Teaching Assistant
- Theoretical Foundations of LLMs, Daniel Hsu
- Machine Learning Theory, Daniel Hsu
- Machine Learning, Sanjoy Dasgupta
- Mathematics for Algorithms and Systems, Russell Impagliazzo
- Software Project for Computer Architecture, Hung-Wei Tseng
- Introduction to Programming, Gerald Soosairaj
Grader
- Abstract Algebra
- Differential Equations
- Mathematical Reasoning
Reviewing
- COLT 2025, 2026
- ICML 2025
- NeurIPS 2024, 2025
Service
- Volunteer, Pre-Submission Application Review (PAR) Program, which provides feedback on statements and CVs from students applying to the Columbia CS PhD program.
Miscellaneous
I am deeply grateful to my advisors and many others in the community for their great mentorship, including Bingbin Liu, Surbhi Goel, Samuel Deng, Yuhao Li, Hantao Yu, Clayton Sanford, Navid Ardeshir, Samantha Chen, Geelon So, Ezra Edelman, Sihan Liu and this list goes on and on… I would like to pass on this culture, so feel free to reach out to me if you think I might be helpful!
I like playing ping pong, climbing, skiing and karaoke :)