Human role

ML Infra Engineer, Modeling

Physical Intelligence

San Francisco aijobs Today
Apply now

The role

Job description

Build experiment launch abstractions; Debug experiments; Debug performance bottlenecks; Design training infrastructure; Implement checkpointing; Implement metrics logging; Implement training infrastructure; Improve device utilization; Improve throughput; Maintain training infrastructure; Manage job orchestration; Monitor experiments; Optimize training performance; Profile memory usage; Reproduce experiments; Scale distributed training; Schedule training jobs; Translate research needs into infras

View full posting

Index terms

Skills & keywords

CheckpointingCloud ComputingData loadingDistributed TrainingExperiment trackingGCPGKEGPUJAXKubernetesMachine LearningMetrics logging

Get roles like this in your inbox

New agentic AI jobs, curated every Thursday. No spam.

Explore more

All General AI jobs
Physical IntelligenceML Infra Engineer, Modeling
Apply