ML Infra Engineer (TPU/Jax/Optimization)

Physical Intelligence(22 days ago)

HybridFull TimeSenior$95,000 - $130,000Engineering

About this role

This role involves developing and maintaining scalable training infrastructure for large-scale ML models, working closely with researchers and engineers to optimize and manage training systems across cloud and hardware platforms.

View Original Listing

Required Skills

JAX
PyTorch
Distributed Training
GCP
Kubernetes
TPU
GPU
Data Loaders
Performance Optimization
Model Training

About Physical Intelligence

physicalintelligence.company

Physical Intelligence (π) is at the forefront of integrating general-purpose AI into the physical realm. Comprising a diverse team of engineers, scientists, and roboticists, the company develops advanced foundation models and learning algorithms for today's robots and future physically-actuated devices. By focusing on real-time action chunking, knowledge insulation, and efficient training methods, Physical Intelligence aims to enhance robotic capabilities and generalize their actions to adapt to various environments and tasks. The company is backed by prominent investors, including Jeff Bezos and OpenAI, underscoring its potential impact on AI-driven robotics.

View more jobs at Physical Intelligence →