Senior Software Engineer, ML Platform

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 is an innovative biotech startup focused on leveraging AI to transform drug discovery and development. The company is leading the development of generative AI models to design protein and antibody therapeutics, enabling the creation of medicines against historically hard-to-drug molecular targets. It is also developing foundation models for biology and disease to enable better target elucidation and patient stratification. Collectively, these technologies aim to continually enable the identification of novel therapies and to improve success in drug development. Xaira is headquartered in the San Francisco Bay Area, Seattle, and London.

About the RoleWe 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.

What You Will Do

• 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!

Preferred Skills and 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.

Compensation

We offer a competitive compensation and benefits package, seeking to provide an open, flexible, and friendly work environment to empower employees and provide them with a platform to develop their long-term careers. A Summary of Benefits is available for all applicants. We offer a competitive package that includes base salary, bonus, and equity. The base pay range for this position is expected to be $205,000 - $275,000 annually with a total cash compensation up to $235,000 - $325,000; however, the compensation offered may vary depending on the market, job-related knowledge, skills and capabilities, and experience.

Xaira Therapeutics an equal-opportunity employer. We believe that our strength is in our differences. Our goal to build a diverse and inclusive team began on day one, and it will never end.

Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey.Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiringprocess or thereafter. Any information that you do provide will be recorded and maintained in aconfidential file.

As set forth in Xaira Therapeutics’s Equal Employment Opportunity policy,we do not discriminate on the basis of any protected group status under any applicable law.

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection.As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measurethe effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categoriesis as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

• Alcohol or other substance use disorder (not currently using drugs illegally)


• Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, HIV/AIDS


• Blind or low vision


• Cancer (past or present)


• Cardiovascular or heart disease


• Celiac disease


• Cerebral palsy


• Deaf or serious difficulty hearing


• Diabetes


• Disfigurement, for example, disfigurement caused by burns, wounds, accidents, or congenital disorders


• Epilepsy or other seizure disorder


• Gastrointestinal disorders, for example, Crohn's Disease, irritable bowel syndrome


• Intellectual or developmental disability


• Mental health conditions, for example, depression, bipolar disorder, anxiety disorder, schizophrenia, PTSD


• Missing limbs or partially missing limbs


• Mobility impairment, benefiting from the use of a wheelchair, scooter, walker, leg brace(s) and/or other supports


• Nervous system condition, for example, migraine headaches, Parkinson’s disease, multiple sclerosis (MS)


• Neurodivergence, for example, attention-deficit/hyperactivity disorder (ADHD), autism spectrum disorder, dyslexia, dyspraxia, other learning disabilities


• Partial or complete paralysis (any cause)


• Pulmonary or respiratory conditions, for example, tuberculosis, asthma, emphysema


• Short stature (dwarfism)


• Traumatic brain injury

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.