Software Engineer, ML Platform

Develop and improve our model training system, responsible for dispatching distributed training jobs to clusters across multiple clouds. Deploy storage subsystems that improve dataset management and throughput for training datasets. Build evaluation infrastructure that enables easy execution and tracking. Build base tooling for integrating model training with telemetry, experiment tracking, and checkpointing. Collaboration with AI Scientists and engineers to enable GPU-heavy infrastructure for training and inference of biological foundation models; preferred candidates have strong Python, Terraform/Ansible, and deep learning framework skills (Torch/Jax); experience with ML model training infrastructure at scale is required.