Hi! I am a second year Computer Science PhD student at University of Wisconsin–Madison working with Fred Sala. As an undergraduate at Brown University, I was fortunate to work with Stephen Bach.
This summer I'm currently doing a research internship at KRAFTON AI.
My research interests include elastic architectures, Transformers, and data-centric ML. Feel free to reach out for a chat!
Education
University of Wisconsin—Madison
Ph.D. in Computer Science
Advised by: Prof. Fred Sala
Brown University
Sc.B. in Mathematics-Computer Science
Advised by: Prof. Stephen Bach
Publications
Test-Time Scaling Makes Overtraining Compute-Optimal
Nicholas Roberts, Sungjun Cho, Zhiqi Gao, Tzu-Heng Huang, Albert Wu, Gabriel Orlanski, Avi Trost, Kelly Buchanan, Aws Albarghouthi, Frederic Sala
Modern scaling laws.
ICML 2026
Weight Updates as Activation Shifts: A Principled Framework for Steering
Dyah Adila, John Cooper, Alexander Yun, Avi Trost, Frederic Sala
Expressivity of steering.
R&B: Domain Regrouping and Data Mixture Balancing for Efficient Foundation Model Training
Albert Ge, Tzu-Heng Huang, John Cooper, Avi Trost, Ziyi Chu, Satya Sai Srinath Namburi GNVV, Ziyang Cai, Kendall Park, Nicholas Roberts, Frederic Sala
Reclustering data and reusing training gradients for data mixing.
