ML Infra Engineer, Modeling

Role

The ML Infra Engineer, Modeling role focuses on scaling and optimizing training systems and core model code for large-scale JAX-based training. Candidates must have strong software engineering fundamentals and experience building ML training infrastructure, with hands-on experience in JAX or PyTorch.

Location: San Francisco
Employment: unspecified
Work pattern: unspecified
Seniority: unspecified

APPLY ON THE EMPLOYER’S SITE ↗

Source and currency

This vacancy came from the organization’s official Ashby feed. Source update: 2026-09-01T01:09:51.071+00:00. Check the employer’s page for the complete description, current availability and application terms.

• jobs.ashbyhq.com/physicalintelligence/96bf6ea8-fb8d-4a76-bec7-878a8c224cea ↗

Physical Intelligence RESEARCH PROFILE ↗