Senior / Staff Machine Learning Engineer, Applied AI
Lila Sciencesabout 4 hours ago
Cambridge, MA, USA or San Francisco, CA, USASenior / Staff+
H1B Sponsor
Responsibilities
- Close the last-mile gap between Lila AI model capabilities and customer-specific scientific workflows.
- Build evaluation loops that measure model quality, reliability, and customer fit.
- Design experiments to improve model performance across applied customer use cases.
- Feed customer learnings, data signals, and evaluation results back into the Lila AI model improvement cycles.
- Partner with AI researchers to translate model improvements into usable capabilities.
- Work with Software to integrate model behavior into end-to-end product workflows.
- Debug model failures using traces, evaluations, customer context, and scientific feedback.
- Build reusable tooling for model adaptation, evaluation, and deployment 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.