Member of Technical Staff, Machine Learning Engineer

What you'll do

• Translate scientific AI-agent challenges into optimizable problems and build environments for training and deployment.
• Lead large-scale LLM training and develop efficient training, inference, and experimentation infrastructure.
• Work with scientists on scalable, observable, reproducible data pipelines and collaborate with a multidisciplinary research team.

What they ask for

• 5-10+ years of applied ML research and applying ML methods to real-world problems.
• Experience across the ML lifecycle, including data pipelines and provenance, training, deployment, and production validation.
• Fluency in PyTorch, Jax or equivalent framework.
• Demonstrated academic or industry experimentation experience and strong programming expertise across data, ML, and LLM software.

Nice to have

• PhD in Machine Learning, Computer Science, or other quantitative field.
• Familiarity with distributed computing resources or architecting complex distributed systems.

Pay

$240,000 - $350,000 USD base salary

Tools

PyTorch, Jax, LLM

AI focus

Build AI scientist agents and large-scale LLM-based systems for scientific discovery, including training environments and inference infrastructure.