USD 189000-289000/yr

Research Engineer, LLM Post-Training

Cambridge · United States - San Franciscomid

Data ScienceSoftware EngineeringSearch Engineer

0 views0 saves0 applied

Apply Now

Quick Summary

Overview

Your Impact at LILA We’re seeking a Machine Learning Research Engineer specializing in LLM Post Training you’ll design and maintain large-scale training systems,

Technical Tools

Data ScienceSoftware EngineeringSearch Engineer

We’re seeking a Machine Learning Research Engineer specializing in LLM Post Training you’ll design and maintain large-scale training systems, optimize performance for massive models, and integrate cutting-edge techniques to improve efficiency and throughput.

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.

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.

Nice to Have

~1 min read

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

San Francisco, CA or Cambridge, MA (Hybrid and On-Site available depending on team needs).

What We Offer

~1 min read

We offer competitive base compensation with bonus potential and generous early-stage equity. Your final offer will reflect your background, expertise, and expected impact.

What We Offer

~1 min read

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.

LILA combines advanced AI models with proprietary AI Science Factory™ instruments into an operating system for science that executes the entire scientific method autonomously, accelerating discovery at unprecedented speed, scale, and impact across medicine, materials, and energy. Learn more at www.lila.ai.

Guided by our core values of truth, trust, curiosity, grit, and velocity, we move with startup speed while tackling problems of historic importance. If this sounds like an environment you'd love to work in, even if you don't meet every qualification listed above, we encourage you to apply.

Lila Sciences is committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status.

Information you provide during your application process will be handled in accordance with our Candidate Privacy Policy.

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.