Senior ML Platform Engineer - Equity & Biotech Impact

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

Job summary

Xaira Therapeutics is seeking a Senior Software Engineer to design, build, and deploy the AI infrastructure powering its research team. You will enable thousands of GPUs for training and inference of advanced biologically focused models, working across cloud clusters and storage systems.

Ideal candidates have 5+ years building ML systems, strong Python skills, and experience with Terraform/Ansible and deep learning frameworks like Torch or Jax.

Qualifications

• 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 concepts.
• Experience with the infrastructure needs of large-scale model training.

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.
• Collaborate with AI Scientists and other engineers to enable scalable GPU workloads and efficient training pipelines.

Skills

Python

ML systems

Distributed training

Deep learning

Terraform

Ansible

Kubernetes

Torch

Jax

Leadership

Education

Degree in Computer Science / ML / Computational Biology

Tools

Terraform

Ansible

Kubernetes

Git

Job description

Xaira Therapeutics is seeking a Senior Software Engineer to design, build, and deploy the AI infrastructure powering its research team. You will enable thousands of GPUs for training and inference of advanced biologically focused models, working across cloud clusters and storage systems.

Ideal candidates have 5+ years building ML systems, strong Python skills, and experience with Terraform/Ansible and deep learning frameworks like Torch or Jax.