Shaojie Bai
Member Of Technical Staff at Thinking Machines Lab
San Francisco, California, United States
Summary
🤩
Rockstar🎓
Top SchoolShaojie Bai is a machine learning researcher-engineer with 11 years of experience who blends deep academic rigor (PhD in ML from CMU) with applied AI research at scale in industry. He has driven cutting-edge work in self-supervised and multimodal generative models across Meta’s Reality Labs and GenAI teams and now contributes at Thinking Machines Lab, focusing on audio-visual modeling, synthetic data and multimodal LLMs. His doctoral thesis on equilibrium approaches in deep learning under Zico Kolter reflects a strong foundation in novel architectures and optimization, paired with practical wins such as top rankings in real-world ML competitions. Comfortable shifting between research and production, he has a track record of shipping large-scale pretraining experiments and coordinating academic seminars, signaling both leadership and community engagement. Based in San Francisco, he combines theoretical depth with hands-on systems experience, often exploring implicit and sequential modeling perspectives that inform his hybrid research-engineering approach.
11 years of coding experience
5 years of employment as a software developer
High School Diploma General Academics, High School Diploma General Academics at Shenzhen Foreign Languages SchoolDoctor of Philosophy - PhD Machine Learning, Doctor of Philosophy - PhD Machine Learning at Carnegie Mellon University
Github Skills (55)
python10
temporal10
in-progress10
machine-learning10
sequence-modeling10
trellis10
deep-learning10
gpu10
sequence10
modeling10
pytorch9
language-processing9
numpy9
datasets9
acceleration9
Programming languages (5)
Github contributions (5)
locuslab/TCN
Mar 2018 - Mar 2022
Sequence modeling benchmarks and temporal convolutional networks
Contributions:15 commits, 3 PRs, 8 pushes in 4 years 1 month
sequencedeep-learningconvolutionalconvolutional-networkstemporal
locuslab/deq
Sep 2019 - Oct 2021
[NeurIPS'19] Deep Equilibrium Models
Contributions:70 commits, 2 PRs, 67 pushes in 2 years 2 months
pytorchequilibriumdeep-learningneuripsdeep-equilibrium-models
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.