Software Engineer, ML Platform

About the Company

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.

Responsibilities

• Develop and improve the model training system for dispatching distributed training jobs to clusters across multiple clouds
• Deploy storage subsystems to improve dataset management and throughput for training datasets
• Build evaluation infrastructure to enable easy execution and tracking
• Build base tooling for integrating model training with internal infrastructure like telemetry, experiment tracking, and checkpointing

Requirements

• 5+ years of industry experience building and deploying ML systems in production environments
• Strong programming skills in Python
• Experience with infrastructure/ops tools such as Terraform or Ansible
• Experience with deep learning frameworks such as Torch or Jax
• Solid understanding of machine learning
• Experience with the infrastructure needs of large-scale model training

Preferred Qualifications

• Degree in Computer Science, Machine Learning, Computational Biology, or a related field
• Experience leading technical projects and driving cross-functional execution