Senior / Staff Machine Learning Infrastructure Engineer

California Life • South San Francisco, CA

82d • $217,000 - $265,000

About The Position

Calico seeks a Senior / Staff Machine Learning Infrastructure Engineer to productize machine learning algorithms and build a cutting-edge computing and machine learning platform for analyzing biological data sets and modeling macromolecule sequences and structures. You will contribute to a cross-functional effort to create a world-class computing and data analysis platform that supports research initiatives at Calico through the following: * Develop and optimize: Build software infrastructure and tools that accelerate ML research, manage training and inference workflows, and facilitate efficient data ingestion and curation. * Define engineering best practices: Lead efforts to reorganize and refactor existing research codebases to improve code quality and reproducibility. * Collaborate for impact: Work closely with our ML research scientists to deploy models that address critical questions in aging research and advance Calico's mission.

Requirements

• Ph.D. in Computer Science or related technical field plus 4+ years of industry experience; or M.S. in Computer Science or related technical field plus 7+ years of industry experience.
• Strong software engineering skills and substantial expertise in Python.
• Experience in designing, developing, and maintaining ML infrastructure components in a production environment.
• Experience in productionizing ML models developed in Jax, Tensorflow, or Pytorch.
• Proficiency in building data pipelines, model training and deployment systems, and monitoring tools.
• Excellent problem-solving and analytical skills.
• Understanding of distributed systems and parallel computing.
• Track record of outstanding communication and collaboration in a cross-functional environment.
• Desire to stay current with the latest ML infrastructure technologies and best practices.
• Willingness to work onsite at least four days a week.

Nice To Haves

• Domain expertise in computational biology, biochemistry or structural biology.
• Prior experience working with biological data.
• Experience in working with Google3.

Responsibilities

• Develop and optimize software infrastructure and tools for ML research.
• Manage training and inference workflows.
• Facilitate efficient data ingestion and curation.
• Lead efforts to reorganize and refactor existing research codebases.
• Improve code quality and reproducibility.
• Collaborate with ML research scientists to deploy models.

Benefits

• Estimated base salary range of $217,000 - $265,000.
• Eligibility for two annual cash bonuses.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Industry: Insurance Carriers and Related Activities
• Education Level: Ph.D. or professional degree
• Number of Employees: 251-500 employees