Senior ML Engineer, Applied AI — LLM Production

Lila Sciences; San Francisco, Cambridge (CA, MA); On-site; USD 180,000 - 336,000; Full time

Benefits offered by this job

Medical, dental, and vision coverage

Flexible time off

Paid parental leave

Commuter benefits

Company subsidized lunch program

Job summary

Lila Sciences in San Francisco, CA is looking for a Senior/Staff Machine Learning Engineer to enhance AI models tailored to specific customer scientific needs. This position will work closely with AI Researchers and Software teams to develop reliable workflows, ensuring that frontier model capabilities meet real customer contexts.

You will be responsible for debugging model behavior, designing experiments, and collaborating across teams. The expected base salary ranges from $180,000 to $336,000, alongside competitive benefits and equity opportunities.

Qualifications

• Strong experience in building, training, or evaluating machine learning models.
• Strong software engineering skills in Python and modern ML frameworks.
• Experience with distributed ML training frameworks like Megatron-LM.
• Ability to debug model behavior using data and logs.

Responsibilities

• Close the last-mile gap between AI model capabilities and customer workflows.
• Build evaluation loops that measure model quality and reliability.
• Design experiments to improve model performance.
• Partner with AI researchers to translate improvements into usable capabilities.

Skills

Building, training, or evaluating ML models

Software engineering in Python

Experience with PyTorch

Experience with TensorFlow

Distributed ML training frameworks

Designing experiments

Debugging model behavior

Clear communication skills

Tools

PyTorch

TensorFlow

Ray

Job description

Lila Sciences in San Francisco, CA is looking for a Senior/Staff Machine Learning Engineer to enhance AI models tailored to specific customer scientific needs. This position will work closely with AI Researchers and Software teams to develop reliable workflows, ensuring that frontier model capabilities meet real customer contexts.

You will be responsible for debugging model behavior, designing experiments, and collaborating across teams. The expected base salary ranges from $180,000 to $336,000, alongside competitive benefits and equity opportunities.