Principal, Machine Learning Engineer
Lila SciencesSan Francisco, California, United States · Posted 4 months agoDescription
Your Impact at LILA
Lila is building a platform where AI and automation co-evolve to solve the hardest problems in medicine. Within Life Science AI (LSAI), the Foundation Models team builds foundation models that learn across biological sequence, molecular structure, and experimental data to power automated scientific discovery across Lila's life science domains.
We are seeking a Senior or Principal Scientist to set the direction for our work on structure prediction and co-folding. The team's current emphasis is protein–protein and complex prediction in support of antibody and biologics design, and on making those predictions good enough to drive real experimental decisions. You will own models end to end, from problem formulation and architecture through training at scale, evaluation, and integration into Lila's closed-loop discovery engine.
This is a high-impact IC role for someone operating at the frontier of structure-aware generative AI for biology. You will shape the technical agenda for structural foundation model research, collaborate closely with experimental scientists to close the computational–experimental loop, and represent Lila's work to the broader scientific community.
What You'll Be Building
- Drive research on structure prediction and co-folding models for protein complexes, protein–protein interactions, and related biomolecular systems
- Design, train, and evaluate models that advance the state of the art in AlphaFold-style co-folding, diffusion models, protein language models, and related structure-aware ML methods
- Set the evaluation bar for the program, building frameworks that establish model generalization to challenging de novo design problems
- Own training, inference, and evaluation at scale across large GPU clusters
- Shape the end-to-end ML process within Lila's "Lab-in-the-Loop" lifecycle: steer data generation strategy, build pipeline models, and design feedback loops where experimental results improve model performance
- Extend into adjacent foundation model research where it strengthens the structural work, including biological sequence design and multimodal scientific reasoning
- Translate complex biological questions into well-defined ML problems and interpret model outputs in collaboration with wet-lab scientists, structural biologists, and computational biologists
- Advance research standards and methodology within the foundation models program, contributing insights that influence approaches across adjacent teams
- Represent Lila's foundation model research externally through publications at premier venues, conference presentations, and community engagement
What You’ll Need to Succeed
- PhD in Computer Science, Machine Learning, Computational Biology, Biophysics, or a related quantitative field
- Demonstrated ability to formulate and drive research programs independently, from problem definition through publication and deployment
- Fluency across ML and at least one life science domain (structural biology, protein engineering, molecular biology, genomics, or related), with experience designing computational experiments grounded in biological reality
- Strong track record of cross-functional collaboration with experimental scientists, translating between ML and biology
- Expertise in ML frameworks (PyTorch, JAX, or TensorFlow) and experience with large-scale distributed training infrastructure (AWS, GCP, or on-prem clusters)
Bonus Points For
- Strong expertise in structure prediction, co-folding, geometric deep learning, or structure-aware molecular ML, with a track record of training these models
- Experience with AlphaFold or AlphaFold-derived methods (e.g., Boltz, Protenix), RFdiffusion, or protein language models
- Experience in computational protein design, particularly antibody and nanobody engineering
- Strong expertise in generative model architectures and training, with hands-on experience training models on distributed infrastructure
- Experience designing biological sequences or molecular structures with demonstrated wet-lab validation
- Contributions to open-source ML tools, frameworks, or benchmark datasets for scientific applications
- Experience with agentic frameworks or active learning loops in scientific contexts
- Multiple high-impact first-author or senior-author publications, or open-source contributions in AI for Science, at premier venues (NeurIPS, ICML, ICLR, AAAI, Nature Methods, Nature Biotechnology, or equivalent)
Compensation
We offer competitive base compensation with bonus potential and generous early-stage equity. Your final offer will reflect your background, expertise, and expected impact.
U.S. Benefits. 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.
International Benefits. 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.
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.
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.
We’re All In
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.
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.
Similar jobs
Est. 326,000 USD
Your Impact at LILA Lila is building a platform where AI and automation co-evolve to solve the hardest problems in medicine. Within Life Science AI (LSAI), Machine Learning engineers build and operate the systems that tu…
Est. 240,000 USD
Your Impact at Lila Lila is building a platform where AI and automation co-evolve to solve the hardest problems in medicine. Within Life Science AI (LSAI), the Foundation Models team builds foundation models that learn a…
Est. 293,000 USD
Your Impact at LILA Lila Sciences is seeking a Machine Learning Scientist, Cofolding and Structure-Aware ML to train next-generation cofolding models for drug discovery. This role is focused on improving models that reas…
Est. 201,000 USD
Your Impact at LILA We are seeking a Senior Principal Scientist, Small Molecule Therapeutics to lead the protein science and structural biology strategy supporting small molecule therapeutics across Lila’s discovery plat…
Est. 405,000 USD
Your Impact at LILA LSAI (Life Sciences Artificial Intelligence) is building the computational platform that powers Lila's work in the life sciences. We are looking for a Research Engineer to own and build the core codeb…
Est. 276,000 USD
Your Impact at LILA Scientists shouldn't have to context-switch between a dozen tools to go from hypothesis to result. We're building the platform that makes this a reality — and we need engineers who want to solve probl…
Est. 276,000 USD
Your Impact at LILA We are seeking a Senior Principal or Principal Software Engineer, Scientific System of Record to join our Scientific System of Record Team and help define and build the next-generation AI-driven scien…
Est. 192,000 USD
Your Impact at LILA Scientists shouldn't have to context-switch between a dozen tools to go from hypothesis to result. We're building the platform that makes this a reality — and we need engineers who want to solve probl…
Est. 179,300 USD
Your Impact at LILA Lila Sciences is seeking a Computational Biophysics Scientist to help ensure that our AI agents can use biophysics tools in ways that are scientifically correct, reliable, and useful for drug discover…
Est. 276,000 USD
Your Impact at LILA Join us in shaping the future of science! We are seeking Principal Software Engineers with backend experience to join our Data Platform Team (Data), where you’ll collaborate with software engineers, l…
Est. 259,000 USD
Your Impact at LILA Lila Sciences is seeking a Research Engineer, Scientific Computing and ML/Physics Infrastructure to help turn promising research tools into robust, scalable systems. This role bridges research and pro…
Est. 192,000 USD
Your Impact at LILA Join us in shaping the future of science! We are seeking Senior Software Engineers with full stack experience to join our Scientific System of Record Team (SSR), where you’ll collaborate with software…
Est. 239,000 USD
Your Impact at LILA We are growing our Applied AI org and seeking talented Senior/Staff Machine Learning Engineers with expertise in LLM training, evaluation, and production-oriented ML systems. You’ll work on improving…
Est. 216,000 USD
Your Impact at LILA We are seeking a Staff Software Engineer to join our Scientific System of Record Team and help build the next-generation AI-driven scientific platform. You will focus on developing user interfaces, se…
Est. 207,000 USD
Your Impact at LILA We are seeking a Senior Software Engineer to join our Applied AI group and help build the next generation of our AI-driven scientific platform. In this role, you will design and optimize the backend s…
Est. 178,000 USD
Your Impact at LILA The Senior/Principal Scientist, Small Molecule Therapeutics will lead early hit identification efforts, leveraging their expertise in DNA-encoded libraries (DEL) and complementary screening technologi…
Est. 257,000 USD
Your Impact at LILA Lila Sciences is building the infrastructure for autonomous science — AI systems that design experiments, run them in physical labs, and close the loop across life sciences, chemistry, and materials s…
Est. 194,000 USD
Your Impact at LILA We're building an AI-driven drug discovery factory that closes the loop between computational design and wet-lab experiment in days instead of months. The bottleneck for that loop is data: every compo…
Est. 216,000 USD
Your Impact at LILA Join us in shaping the future of science! We are seeking Senior Software Engineers with backend experience to join our Data Platform Team (Data), where you’ll collaborate with software engineers, lab…
Est. 220,000 USD
Your Impact at LILA Lila Sciences is hiring a Director of Product, Life Sciences (Chemistry): a strategic, cross-functional product leader who shapes the vision, roadmap, and execution for platforms that accelerate the d…
Est. 232,000 USD
Your Impact at LILA The Staff/Principal DevOps Engineer will drive the design, implementation, and optimization of our infrastructure and delivery platforms. This role bridges platform engineering, site reliability, and…
Est. 345,000 USD
Your Impact at LILA As Lila scales its AI Science Factories globally, the VP of Engineering will create and sustain an engineering model for a unified, scalable platform for autonomous experimentation. This role will be…
Est. 177,000 USD
Your Impact at LILA Join us in shaping the future of science! We are seeking Senior Software Engineers with backend experience to join our Lab Software Team (LaS), where you’ll collaborate with software engineers, lab sc…
Est. 194,000 USD
Your Impact at LILA Lila Sciences is the world’s first scientific superintelligence platform and autonomous lab for life, chemistry, and materials science. We are pioneering a new age of boundless discovery by building t…
Est. 255,000 USD
Your Impact at LILA Scientists shouldn't have to context-switch between a dozen tools to go from hypothesis to result. We're building the platform that makes this a reality — and we need engineering leaders who want to b…
Est. 345,000 USD
Your Impact at LILA Lila is seeking a highly motivated and experienced engineering leader to lead a team responsible for our Lila's product data platform. You will own the data platform and infrastructure end-to-end — ar…
Est. 301,500 USD
Your Impact at LILA As a Principal Security Engineer focused on AI Security, you will define and drive the technical strategy for securing how AI is used across Lila's enterprise. You will operate as a senior individual…
Est. 293,000 USD
Your Impact at LILA Lila Sciences is seeking a Machine Learning Scientist, Data-Efficient Learning for Drug Discovery to build models and learning strategies for settings where data is scarce, expensive, and intentionall…
Est. 237,000 USD
Your Impact at LILA Lila Sciences is seeking a Principal Software Engineer, Instrument Simulations to architect and lead the development of our next-generation laboratory automation platform. In this role, you'll be resp…
Est. 179,300 USD
Your Impact at LILA Lila Sciences is seeking a Scientist, Computational Chemistry, Drug Discovery to help guide, evaluate, and improve AI-driven drug discovery workflows. This person will bring practical computational ch…