Senior Software Engineer, ML Platform

Xaira Therapeutics is seeking a Software Engineer to join their Platform team to build and deploy AI infrastructure for drug discovery. You will collaborate with AI Scientists and engineers to optimize GPU usage for training biological foundation models, working on MLOps, backend APIs, and distributed training systems.

• Develop and improve model training systems for distributed jobs across multiple clouds
• Deploy storage subsystems for improved dataset management and throughput
• Build evaluation infrastructure for easy execution and tracking
• Create base tooling for integrating model training with telemetry and experiment tracking
• 5+ years of industry experience building and deploying ML systems
• Strong Python skills and experience with infrastructure tools like Terraform