Senior Software Engineer, ML Platform
Join Xaira Therapeutics as a Senior Software Engineer to build AI infrastructure for drug discovery and development.
Posted by employer 3 months ago
First seen on Joblaze 1 week ago
Last verified on the company career page 1 day ago
AI in the day-to-day
Xaira leverages AI to transform drug discovery and development through generative AI models and biological foundation models.
Requirements
- Experience
- 5+ years
- Education
- Bachelor's degree
Not disclosed in this posting: visa sponsorship.
Benefits
Joblaze summary
In this role, the Senior Software Engineer will focus on designing and deploying the AI infrastructure that supports Xaira Therapeutics' research initiatives, particularly in managing distributed training jobs across multiple cloud environments. Key skills include expertise in MLOps, backend engineering, and familiarity with tools like Kubernetes and deep learning frameworks such as Torch or Jax. This position is ideal for someone with over five years of experience in building ML systems and a collaborative mindset, though prior biology knowledge is not a prerequisite. Xaira's innovative environment emphasizes the integration of cutting-edge AI technologies in drug discovery.
Joblaze insights
- Listed 9 days ago on Joblaze — tracked directly from Xaira Therapeutics's career page.
- Xaira Therapeutics has 23 other open roles including Principal Scientist, Biology, Senior/ Staff AI Research Engineer, Lab Software Engineer.
- 6517 active Python roles on Joblaze right now.
- 1134 active Terraform roles on Joblaze right now.
- 261 active Ansible roles on Joblaze right now.
- Salary band is in line with the typical range for AI/ML roles (median ~$188,000).
- Pays more than 73% of the 3224 salaried Python roles we track (median $160,000). See Python salary trends
Quick facts
- Is the Senior Software Engineer, ML Platform role remote?
- It's hybrid — Xaira Therapeutics expects some on-site time in Seattle, Washington, United States.
- What's the salary range?
- Xaira Therapeutics lists $189,000–$245,000 for this role.
- How much experience is required?
- At least 5 years of relevant experience for this Senior Software Engineer, ML Platform role.
- Where is the role based?
- Xaira Therapeutics is hiring for this position in Seattle, Washington, United States.
- What's the tech stack?
- Joblaze extracted these technologies from the posting: Ansible, JAX, Python, Terraform, Torch.
- What seniority level is this role?
- Xaira Therapeutics targets senior candidates for this position.
- Is this full-time or contract?
- Full-time for this Senior Software Engineer, ML Platform role at Xaira Therapeutics.
From the original posting
About Xaira Therapeutics
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 Role
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.
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 $189,000 - $245,000 annually; however, the base pay 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.
TO ALL RECRUITMENT AGENCIES: Xaira Therapeutics does not accept agency resumes. Please do not forward resumes to our jobs alias or employees. Xaira Therapeutics is not responsible for any fees related to unsolicited resumes.