---
title: "ML Infrastructure Engineer at PI – San Francisco, CA"
canonical: "https://www.metaintro.com/job/ml-infrastructure-engineer-at-pi-a9698a68-e28f-5ec8-a581-dbdc52d4d204"
language: "en"
author: "Metaintro"
---

[/company/pi](/company/pi)[PI](/company/pi)
# ML Infrastructure Engineer at PI

Department: Machine Learning

**Setup**
: On-site

**Location**
: San Francisco, California

**Level**
: not_specified

**Posted**
: Sep 1, 2026

## Description

Physical Intelligence is hiring an ML Infrastructure Engineer to build and operate large-scale model-training infrastructure. The role owns GPU/TPU compute, job orchestration, checkpointing, metrics, distributed JAX training, performance optimization, experiment reproducibility, and core training code. It partners with researchers and model engineers to translate research ideas into production training runs and requires strong software engineering, distributed-training, cloud-workload, and debugging experience.

## Trending job searches

Every query opens live roles, salary samples, and market demand — tap a search to run it instantly.

- [Find ML Infrastructure Engineer jobs in San Francisco, California, United States](/jobs/search?q=Find+ML+Infrastructure+Engineer+jobs+in+San+Francisco%2C+California%2C+United+States)
- [Find roles similar to ML Infrastructure Engineer in San Francisco, California, United States](/jobs/search?q=Find+roles+similar+to+ML+Infrastructure+Engineer+in+San+Francisco%2C+California%2C+United+States)
- [Find ML Infrastructure Engineer jobs with JAX and Distributed training](/jobs/search?q=Find+ML+Infrastructure+Engineer+jobs+with+JAX+and+Distributed+training)
- [Find jobs similar to ML Infrastructure Engineer](/jobs/search?q=Find+jobs+similar+to+ML+Infrastructure+Engineer)
- [Find ML Infrastructure Engineer jobs posted this week](/jobs/search?q=Find+ML+Infrastructure+Engineer+jobs+posted+this+week)
- [Find Developer jobs in San Francisco, California, United States](/jobs/search?q=Find+Developer+jobs+in+San+Francisco%2C+California%2C+United+States)
- [Find Developer jobs with JAX and Distributed training](/jobs/search?q=Find+Developer+jobs+with+JAX+and+Distributed+training)
- [Find remote Developer jobs](/jobs/search?q=Find+remote+Developer+jobs)

## Get More from Metaintro

Unlock powerful job search, personalized recommendations, and deep career insights from comprehensive, market-leading data and live market signals.

- [Browse all jobs](/jobs/search)
- [Access help center](/help)
- [Explore our content](/blog)

![](https://cdn.metaintro.com/rs:fill:1200:800/q:30/plain/images/bridges/bridge-expand.1df895c6bd76d96f.png)

For job seekers

## Ready to find a role that actually fits?

Upload your résumé, start a Job Search Thread, and let Metaintro rank real openings against your experience — then guide you from search to offer.

[Get Started Free](/signup)[Search matching jobs](/jobs/search)

Match

Compare live roles against your current evidence.

Position

Turn proof projects into role-specific applications.

Improve

Use market feedback to keep the skill plan current.

[Return to navigation](#main-navigation)