Senior ML Platform Engineer: Distributed GPU Training Infra

Xaira Therapeutics, Inc.; Seattle (WA); On-site; USD 189,000 - 245,000; Full time

Job summary

Xaira Therapeutics is seeking a Senior Software Engineer to join our Platform team to design, build, and deploy the AI infrastructure that powers our world-class research. You’ll work with AI Scientists and engineers to enable thousands of GPUs for training and inference of biology foundation models in Seattle.

This role covers MLOps for GPU clusters and backend API work, with opportunities to dive into storage, telemetry, and experiment tracking.

Qualifications

• Degree in Computer Science, ML, computational biology or related field.
• 5+ years of industry experience building and deploying ML systems.
• Experience leading technical projects and cross-functional execution.
• Strong programming skills in Python.

Responsibilities

• Develop and improve our model training system, 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.
• You’ll get to go in the lab; this can be a perk for the right candidate.

Skills

Python

Machine Learning

Leadership

Cross-functional collaboration

Problem-solving

Education

Bachelor's degree in CS/ML/Computational Biology or related field

Tools

Terraform

Ansible

Torch

Jax

Job description

Xaira Therapeutics is seeking a Senior Software Engineer to join our Platform team to design, build, and deploy the AI infrastructure that powers our world-class research. You’ll work with AI Scientists and engineers to enable thousands of GPUs for training and inference of biology foundation models in Seattle.

This role covers MLOps for GPU clusters and backend API work, with opportunities to dive into storage, telemetry, and experiment tracking.