Machine Learning Systems Architect at Achira

Achira is hiring a Machine Learning Systems Architect to design and maintain scalable, asynchronous workflows for data generation, training, and evaluation of foundation simulation models. The role focuses on rationalizing machine-learning systems architecture, solving blockers at foundation-model scale, and connecting research scientists with infrastructure teams. Candidates should have at least two years of relevant industry experience, strong PyTorch and JAX skills, familiarity with asynchronous primitives and library design, and a track record of well-documented code. The role is hybrid in San Francisco or New York City, with travel to San Francisco as needed.