Senior ML Engineer I/II, Agentic Simulations

Your Impact at Lila

As a member of our team in the Physical Sciences division, you will shape the ML- and agent-driven infrastructure that enables Lila’s scientific superintelligence to autonomously construct, execute, and interpret complex physics simulations. Your work will focus on building the core systems that allow AI agents to reason over and control computational experiments, spanning from electronic structure calculations to surrogate-driven atomistic modeling, and beyond. Your work will directly influence how autonomous computational processes will explore chemical and materials landscapes with unprecedented autonomy and robustness.

What You'll Be Building

• Architect and implement agentic frameworks that support dynamic, multi-stage simulation workflows for scientific tasks
• Develop pipelines enabling agents to autonomously plan, schedule, execute, and interpret computational tasks at scale
• Build integration layers and APIs that connect ML models, large-scale simulation engines, databases, and heterogeneous compute platforms.
• Work with AI researchers to productionize agent behaviors, including tool-use strategies, simulation-aware decision loops, and adaptive task planning.
• Improve the robustness, modularity, performance, and reproducibility of agent-driven computational workflows; build internal tooling for debugging, observability, and validation.

What You’ll Need to Succeed

• MS/PhD or equivalent experience in Computer Science, ML/AI, Scientific Computing, or a related technical field.
• Strong experience building ML-driven pipelines, workflow systems, or tool-use frameworks, ideally for complex or scientific applications.
• Proficiency in Python and ML ecosystems; experience with one or more compiled languages (e.g., C++, Rust, Julia) is beneficial.
• Familiarity with large-scale scientific or engineering software, including integrating external tools into automated computational workflows.
• Experience with distributed systems, HPC environments, cloud platforms, or accelerator-based computing.
• Deep understanding of modern ML architectures and their deployment in production systems (e.g., GNNs, transformers, diffusion models, multimodal or tool-using models).
• Strong engineering fundamentals: reproducibility, testing, modular design, CI/CD, and scalable ML operations.

Bonus Points For

• Experience developing or integrating agentic frameworks, autonomous ML pipelines, or multi-step tool-using agents, particularly for scientific applications
• Background in large-scale simulation frameworks, scientific workflow orchestration, or automated computational experiment platforms.
• Contributions to open-source ML infrastructure, workflow engines, agent frameworks, or scientific software ecosystems.
• Familiarity with data-centric engineering practices: streaming systems, provenance tracking, distributed metadata services, or large-scale data orchestration.
• Experience with workflow orchestration systems (e.g., Flyte, Prefect, Airflow) and container orchestration (Kubernetes).
• Familiarity with materials science simulation codes (VASP, LAMMPS) or workflow frameworks (atomate, aiida) is a plus but not required.
• Expertise in advanced ML techniques relevant to agentic reasoning (planning models, self-improving systems, multi-tool agents, RLHF/RLAIF workflows).

About 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 the capabilities to apply AI to every aspect of the scientific method.  We are introducing scientific superintelligence to solve humankind's greatest challenges, enabling scientists to bring forth solutions in human health, climate, and sustainability at a pace and scale never experienced before. Learn more about this mission at  www.lila.ai

If this sounds like an environment you’d love to work in, even if you only have some of the experience listed below, we encourage you to apply.

Compensation

We expect the base salary for this role to fall between $176,000 –$304,000  USD per year, along with bonus potential and generous early equity. The final offer will reflect your unique background, expertise, and impact.

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.