Senior ML Platform Engineer for Infra in Drug Discovery

Xaira; Seattle (WA); On-site; USD 205,000 - 275,000; Full time

Job summary

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

This role spans MLOps of GPU clusters, backend API work, and cross-cloud orchestration.

Qualifications

• 5+ years of industry experience building and deploying ML systems in production.
• Degree in CS/ML/Computational Biology or related field.
• Strong Python programming abilities.
• Experience leading technical projects and cross-functional work.

Responsibilities

• Develop and improve model training system to dispatch distributed jobs across clusters.
• Deploy storage subsystems to improve dataset management and throughput.
• Build evaluation infrastructure for easy execution and tracking.
• Create tooling to integrate training with telemetry, experiment tracking, and checkpoints.

Skills

Python

ML systems

Team leadership

Problem solving

Education

Degree in Computer Science, Machine Learning, Computational Biology, or a related field

Tools

Terraform

Ansible

Torch

JAX

Kubernetes

Job description

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

This role spans MLOps of GPU clusters, backend API work, and cross-cloud orchestration.