Mark Lee
Summary
🤩
RockstarMark Lee is an AI research scientist with 11 years of experience building production-grade ML systems, currently working at Meta's Superintelligence Lab. He was the first member of Apple's Foundation Models team under Ruoming Pang and led the development of AXLearn, bridging open-source research tooling with large-scale model engineering. At Uber ATG he built and optimized core remote inference and perception infrastructure—rewriting services and clients to simplify APIs and doubling inference throughput. Combining research rigor from a CMU teaching stint with hands-on backend and Golang expertise, he excels at turning foundation-model research into reliable, reproducible production platforms. Based in the San Francisco Bay Area, he focuses on scalable training, inference platforms, and tooling that accelerates research-to-production cycles.
11 years of coding experience
Github Skills (23)
deep-learning10
jax10
pytorch9
autodiff9
python9
machine-learning9
natural-language-processing9
composable9
gpu9
automatic-differentiation9
jit9
speech-recognition9
nlp9
transformer8
bert8
Programming languages (2)
Github contributions (5)
apple/axlearn
Jul 2023 - Apr 2025
An Extensible Deep Learning Library
Contributions:766 reviews, 428 PRs, 61 pushes in 1 year 8 months
deep-learningjax
markblee/axlearn
Jul 2023 - Apr 2025
Contributions:229 pushes, 411 branches in 1 year 8 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.