About the job
We are seeking a Software Engineer to join our Platform team to design, build, and deploy the AI infrastructure that powers our world-class research team. In this role, you’ll collaborate closely with AI Scientists and other engineers to enable the effective use of thousands of GPUs for training and inferencing cutting-edge biological foundation models.
Responsibilities
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 other internal infrastructure, such as telemetry, experiment tracking, and checkpointing.
Qualifications
Minimum
No minimum qualifications listed.
Preferred
Degree in Computer Science, Machine Learning, Computational Biology, or a related field.
5+ years of industry experience building and deploying ML systems in production environments
Experience leading technical projects and driving cross-functional execution.
Strong programming skills in Python.
Experience with infrastructure/ops tools such as Terraform, Ansible.
Experience with deep learning frameworks such as Torch, Jax.
Solid understanding of machine learning
Experience with the infrastructure needs of large-scale model training.
Strong problem-solving skills and ability to work in a collaborative, multidisciplinary environment.