Senior / Staff Machine Learning Engineer, Applied AI

Highlights

• Turn frontier AI capabilities into reliable workflows for real scientific use cases.
• Close the last-mile gap between Lila models and customer science workflows.
• Help build autonomous science operating system to accelerate discovery at scale.

What they offer

• Competitive base compensation with bonus potential and early-stage equity
• Medical, dental and vision coverage plus employer-paid life and disability insurance (U.S.)
• Flexible time off with generous company-wide holidays and paid parental leave
• Educational assistance program, commuter benefits and subsidized lunch (U.S.)

Ideal candidate

Research-minded engineer who loves bridging models and real-world science. You thrive debugging model behavior, iterating with evaluations, and partnering across teams to ship reliable AI into customer workflows.

Requirements

• Strong experience building, training, adapting, or evaluating machine learning models
• Strong software engineering skills in Python and modern ML frameworks such as PyTorch, JAX, or TensorFlow
• Experience with distributed ML training frameworks (Megatron-LM, TorchTitan, DeepSpeed, Ray)
• Experience designing experiments, evaluation metrics, or test sets for model performance
• Ability to debug model behavior using data, traces, logs, and qualitative feedback
• Experience working across research and engineering teams to move ML capabilities into usable systems
• Familiarity with large language models, multi-modal models, or agentic AI systems
• Clear communication skills for translating customer needs into technical model improvements