Machine Learning Engineer, Distributed & Scalable Training jobs in United States
cer-icon
Apply on Employer Site
company-logo

Lila Sciences · 1 week ago

Machine Learning Engineer, Distributed & Scalable Training

positionCambridge, MA
timeFull-time
remoteHybrid
seniorityMid, Senior Level
money$116K/yr - $170K/yr

Lila Sciences is the world’s first scientific superintelligence platform and autonomous lab for life, chemistry, and materials science. They are seeking a Machine Learning Engineer who will design and maintain large-scale training systems, optimize performance for massive models, and integrate cutting-edge techniques to improve efficiency and throughput.

Artificial Intelligence (AI)Foundational AILife ScienceSoftware
check
H1B Sponsor Likelynote

Responsibilities

Ray-based distributed training infrastructure for LLMs and multi-modal models
Performance optimizations for large-scale model training including training and optimization workflows (SFT, MoE, long-context scaling)
Orchestrate frontier and open source LLMs along with complex compute-intensive tool use
Scalable pipelines for data preprocessing and experiment orchestration, including tools for efficient data loading, pipeline parallelism, and optimizer tuning
System-level performance benchmarks and debugging utilities

Qualification

Distributed ML training frameworksPythonLarge-scale model trainingCloudHPC environmentsC++ kernel contributionsScientific datasets experienceOpen-source ML contributions

Required

Proven experience with distributed ML training frameworks (Megatron-LM, TorchTitan, DeepSpeed, Ray)
Strong software engineering skills (Python, C++ kernel contributions are a plus)
Understanding of large-scale model training techniques
Experience with cloud or HPC environments

Preferred

Prior work with scientific datasets or domain-specific modeling
Contributions to open-source ML frameworks

Benefits

Bonus potential
Generous early equity

Company

Lila Sciences

twittertwittertwitter
company-logo
Lila Sciences creates a scientific superintelligence platform and autonomous labs for life sciences, chemistry, and materials science. It is a sub-organization of Flagship Pioneering.
dateFounded in 2023
location
Cambridge, Massachusetts, USA
people-size201-500 employees
web-link
https://www.lila.ai

H1B Sponsorship

Lila Sciences has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (8)

Funding

Current Stage
Growth Stage
Total Funding
$550M
Key Investors
NVenturesFlagship Pioneering
2025-10-14Series A· $115M
2025-09-14Series A· $235M
2025-03-10Seed· $200M

Recent News

Axios
2025's AI-fueled scientific breakthroughs
2026-01-03
masslive
Here’s what happened to Mass. startup investing in 2025
2025-12-17
MIT Technology Review
AI materials discovery now needs to move into the real world
2025-12-15
Company data provided by crunchbase