Senior Software Engineer, ML Platform

The role

You will design, build and deploy the AI infrastructure used by the research team. You will work with AI scientists and engineers to train and run biological foundation models on large GPU clusters.

What you will do

• Develop and improve the model training system that dispatches 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 to integrate model training with telemetry, experiment tracking and checkpointing.

Required 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.
• Experience with the infrastructure needs of large-scale model training.
• Strong problem-solving skills and ability to work in a collaborative, multidisciplinary environment.

Benefits

• Competitive compensation and benefits package.
• Base salary, bonus and equity.
• Summary of Benefits available to all applicants.
• Open, flexible and friendly work environment.

Before you apply

• Prior experience with biology is not required; the company will teach what is needed.

About the company

Xaira Therapeutics is a biotech startup using AI to transform drug discovery and development. It builds generative AI models for protein and antibody therapeutics and foundation models for biology and disease.