jobs

ML Infra Engineer, Modeling

Physical Intelligence is bringing general-purpose AI into the physical world. We are a group of engineers, scientists, roboticists, and company builders developing foundation models and learning algorithms to power the robots of today and t

Company

Physical Intelligence

Location

San Francisco

Status

Open

Posted

2026-09-01T01:09:51.071+00:00

Physical Intelligence is hiring a mid-level ML Infrastructure engineer for its ML Infrastructure team in San Francisco, on-site. The role focuses on scaling and optimizing model-training systems and main model code. Work includes managing GPU/TPU compute, coordinating jobs, checkpoints, logging, JAX or PyTorch distributed training, performance profiling, and abstractions for experiments. The posting asks for solid software engineering basics, experience with large-scale training infrastructure, training workloads on cloud platforms, distributed systems, and communication across teams. It stands out for close collaboration with researchers on robotics and foundation models.