ML Infra Engineer: Scale & Optimize Large-Scale Training

Physical Intelligence in San Francisco is seeking a hands-on ML infrastructure engineer to scale and optimize our training systems. You will own large-scale training infra, manage GPU/TPU compute, job orchestration, checkpointing, and memory profiling, building efficient JAX pipelines that turn research ideas into production runs. You'll work with researchers and model engineers to push the limits of scalable ML, contribute to core training code, and help ensure reliability and speed across