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.
