ML Systems Architect for Scalable Drug Discovery Workflows

Achira; San Francisco (CA); Hybrid; USD 120,000 - 160,000; Full time

Job summary

Achira is looking for a specialist in machine learning systems architecture to help architect future molecular machine learning solutions. The role involves building robust workflows and collaborating with research scientists. Candidates should have at least two years of industry experience, be highly fluent in PyTorch and JAX, and possess a strong understanding of scalable ML systems. While preferred to work in San Francisco, skilled candidates may work remotely with some travel. Achira values collaboration and innovation.

Qualifications

• At least two years relevant industry experience.
• Highly fluent in and enthusiastic about PyTorch and JAX.
• Able to think in asynchronous primitives.
• Strong views on library design: clean abstractions, minimal surface area, consistency.
• Solid track record of observable artifacts showing clear, well-documented code.
• ML generalist who understands scalable, reliable ML systems.

Responsibilities

• Build and maintain robust multi-stage asynchronous workflows for data generation, training, and evaluations.
• Rationalize machine learning systems design and software architecture.
• Identify blockers and develop scalable solutions.
• Act as a liaison between research scientists and the infrastructure team.

Skills

PyTorch

JAX

Asynchronous programming

Library design

ML systems design

Job description

Achira is looking for a specialist in machine learning systems architecture to help architect future molecular machine learning solutions. The role involves building robust workflows and collaborating with research scientists. Candidates should have at least two years of industry experience, be highly fluent in PyTorch and JAX, and possess a strong understanding of scalable ML systems. While preferred to work in San Francisco, skilled candidates may work remotely with some travel. Achira values collaboration and innovation.