Scientist II / Senior ML Scientist, Cofolding and Structure-aware ML

Lila Sciences Lila Sciences · AI Frontier · Alewife, Cambridge, MA +1 · Physical Sciences AI

Seeking a Machine Learning Scientist to train next-generation cofolding models for drug discovery, focusing on improving models that reason over proteins, ligands, and experimental data. The role involves developing models that learn from DEL and related datasets, connecting molecular and protein context to enhance AI-driven discovery decisions.

What you'd actually do

  1. Train and evaluate cofolding models for protein-ligand and related molecular discovery applications.
  2. Use contrastive learning, representation learning, self-supervised learning, or related methods where they help improve cofolding models trained on molecules, proteins, structures, and experimental readouts.
  3. Develop modeling approaches that make DEL data more useful for learning binding, enrichment, selectivity, and structure-activity signals.
  4. Build and evaluate models informed by Boltz, AlphaFold-style cofolding, equivariant GNNs, and related structure-aware ML methods.
  5. Design training objectives, including contrastive, self-supervised, or multimodal objectives, that connect ligands, proteins, structures, assays, simulations, and experimental data.

Skills

Required

  • PhD or equivalent experience in machine learning, computational biology, computational chemistry, bioinformatics, computer science, or a related field.
  • Hands-on experience training deep learning models for molecular, protein, structural biology, or scientific data applications.
  • Experience with contrastive learning, representation learning, self-supervised learning, or multimodal learning.
  • Familiarity with DEL or related selection, enrichment, screening, or molecular assay datasets.
  • Experience with protein-ligand modeling, cofolding, structure prediction, geometric deep learning, or structure-aware molecular ML.
  • Practical experience with PyTorch, JAX, or an equivalent ML framework.
  • Ability to design careful experiments, ablations, and evaluations for scientific ML models.
  • Strong understanding of data quality, leakage risks, negative construction, and benchmark design.
  • Ability to collaborate across ML, data, computational science, and drug discovery functions.

Nice to have

  • Hands-on experience with DEL data.
  • Drug discovery experience, especially in protein-ligand modeling or molecular optimization contexts.
  • Experience with Boltz, AlphaFold or AlphaFold-derived methods, equivariant GNNs, diffusion models, protein language models, or molecular encoders.
  • Experience training or extending cofolding, protein-ligand, protein-protein, structure prediction, diffusion, or geometric deep learning models.
  • Experience with distributed model training and large-scale scientific data pipelines.
  • Familiarity with active learning or closed-loop molecular design.
  • Experience integrating ML models into agentic scientific workflows.

What the JD emphasized

  • direct experience training modern scientific ML models
  • rigorous evaluation frameworks
  • strong understanding of data quality, leakage risks, negative construction, and benchmark design

Other signals

  • training next-generation cofolding models for drug discovery
  • improving models that reason over proteins, ligands, binding context, and experimental data
  • develop models that learn from DEL and related datasets
  • connect molecular and protein context
  • improve AI-driven discovery decisions