Machine Learning Engineer, Distributed & Scalable Training
What You'll Need to Succeed
- Strong software engineering skills in Python; C++/CUDA a plus
- Experience with distributed ML training frameworks (Megatron-LM, TorchTitan, DeepSpeed, Ray)
- Understanding of large-scale model training techniques for 100B+ models
- Experience with cloud or HPC environment
- Ability to communicate technical results to internal and external stakeholders
Bonus Points For
- Prior work with large scale scientific datasets or domain-specific modeling
- Contributions to open-source ML frameworks
- Experience with RL post-training (RLHF, GRPO, tool-augmented RL)
- Experience training MoE architectures
- We offer competitive base compensation with bonus potential and generous early-stage equity.
- Your final offer will reflect your background, expertise, and expected impact.
- Full-time U.S. employees receive a comprehensive benefits program including medical, dental, and vision coverage; employer-paid life and disability insurance; flexible time off with generous company wide holidays; paid parental leave; an educational assistance program; commuter benefits, including bike share memberships for office based employees; and a company subsidized lunch program.
- Full-time employees outside the U.S. receive a comprehensive benefits program tailored to their region. USD salary ranges apply only to U.S.-based positions; international salaries are set to local market.
Location
San Francisco, CA or Cambridge, MA (Remote, Hybrid, and On-Site available depending on team needs).
Compensation
U.S. Benefits
International Benefits
Expected Base Salary Range
$189,000 - $289,000 USD
About LILA
Lila Sciences is building Scientific Superintelligence™ to solve humankind's greatest challenges. We believe science is the most inspiring frontier for AI. Rather than hard-coding expert knowledge into tools, LILA builds systems that can learn for themselves.
Guided by our core values of truth, trust, curiosity, grit, and velocity, we move with startup speed while tackling problems of historic importance.
A Note to Agencies
Lila Sciences does not accept unsolicited resumes from any source other than candidates. The submission of unsolicited resumes by recruitment or staffing agencies to Lila Sciences or its employees is strictly prohibited unless contacted directly by Lila Science’s internal Talent Acquisition team. Any resume submitted by an agency in the absence of a signed agreement will automatically become the property of Lila Sciences, and Lila Sciences will not owe any referral or other fees with respect thereto.