Andrew Bai
AI Scientist at Mistral AI
Los Angeles, California, United States
Summary
👤
Senior🎓
Top SchoolAndrew Bai is an AI scientist and fifth-year PhD candidate at UCLA with nine years of industry and research experience focused on memorization and forgetting in machine learning, particularly in post-training for LLMs. He has helped demystify why RLHF reduces forgetting compared to supervised fine-tuning and developed practical methods—like computationally-free rehearsal and early-stopping metrics—that improve alignment efficiency and downstream DPO performance. His internships at DeepMind, Google, and Amazon produced high-impact prototypes and benchmarks (e.g., a tool-calling prototype with sub-1s latency and a pedagogically inspired response-format benchmark), showing he bridges rigorous research with production-minded systems. Comfortable across reward modeling, diffusion memorization, long video generation, and LLM agents, he thrives on collaborative problem solving and cross-disciplinary projects. Based in Los Angeles and now at Mistral AI, he pairs deep theoretical insight with hands-on engineering to make alignment research more scalable and practical.
9 years of coding experience
3 years of employment as a software developer
Bachelor's degree, Computer Science, Bachelor's degree, Computer Science at National Taiwan UniversityUniversity of California, Los Angeles
English, Chinese
Github Skills (22)
ipad9
prediction8
generative8
ipados8
image-synthesis7
machine-learning-algorithms7
mosh7
scp6
generative-adversarial-network6
shell5
swift5
vscode5
blink5
deep-learning5
machine-learning5
Programming languages (4)
Github contributions (5)
jybai/TCRI-Benchmark
May 2020 - Dec 2024
Contributions:9 pushes, 1 branch, 3 comments in 4 years 7 months
jybai/jybai.github.io
Jan 2022 - Oct 2024
Contributions:10 pushes, 1 branch in 2 years 10 months
Find and Hire Top DevelopersWe’ve analyzed the programming source code of over 60 million software developers on GitHub and scored them by 50,000 skills. Sign-up on Prog,AI to search for software developers.