Machine Learning Engineer, AI

Highlights

• Build infrastructure connecting frontier AI models and lab systems to cure disease.
• Train biological foundation models on one of the largest AI clusters for science.
• Create agentic systems that connect LLMs with biological foundation models.

What they offer

• Generous employer 401(k) match
• Paid time off to volunteer
• Funding for select family-forming benefits
• Relocation support

Ideal candidate

An infrastructure-minded builder energized by fundamental science who loves partnering with researchers and solving hard systems problems to accelerate discovery and cure disease.

Requirements

• 5+ years building ML systems or distributed infrastructure at scale
• Strong Python, PyTorch, and CUDA skills with custom training loops and understanding of model architectures
• Experience with large-scale distributed systems, multi-GPU training, checkpointing, and failure recovery
• Ability to diagnose performance problems across GPUs, memory, networking, storage, and data loading
• Experience building large-scale data pipelines and shared research tools
• Experience deploying and monitoring models and maintaining automated tests and CI
• Familiarity with Slurm, Docker, and Kubernetes