Senior Software Engineer, ML Platform

Seattle, WAfull timeOn Site

Apply on Employer Site

senior

$189k–$245k

H1B: no

Xairatherapeutics Senior Software Engineer on the Platform team building AI infrastructure for large-scale biological foundation model training. Design and deploy systems for distributed training and dataset management.

Responsibilities

We are seeking a Senior 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.
This role spans a range of problems and skillsets, ranging from MLOps of cutting-edge GPU clusters, to backend engineering of control plane APIs. Our ideal candidate has an opinion about slurm or kubernetes for model training, cares about maximizing bandwidth from the storage subsystems to the GPU, and can build the API paved path for submitting training jobs that are able to dispatch to multiple clusters.
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.
Prior experience with biology is not required - we will teach what you need to know. You’ll get to go in the lab, and for our ideal candidate, this should be a perk, not a chore!

Qualification

Degree in Computer ScienceStrong programming skills in PythonSolid understanding of machine learning

Required

Degree in Computer Science, Machine Learning, Computational Biology, or related field
5+ years of industry experience building and deploying ML systems in production
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

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.

Benefits

Equity/RSU

Company

Xairatherapeutics

View company page

Source: greenhouse

Active within the past hour

More roles at Xairatherapeutics

Lab Automation - Robotics Engineer

South San Francisco, CA

Principal Scientist, Target Discovery & Portfolio Strategy

South San Francisco, CA

Senior Scientist, Antibody Characterization

South San Francisco, CA

Payroll Manager

South San Francisco, CA

Boost Your Interview Chances

Improve Resume Match Score

Free

Your Score

9.0

Strong Applicant

Must-Have Skills for This Role

PythonMLOpsSlurmKubernetesTerraformAnsible

Optimize my Resume

Apply Faster with Autofill

Free

Save your profile once. LumiJobs pre-fills the employer's application form so you stop re-typing the same answers on every site.

Apply With Autofill

Or skip LumiJobs's tools and submit directly on the employer's career site.

Apply on Employer Site