ML Infra Engineer (TPU/Jax/Optimization)

Physicalintelligence

•

Jan 23, 2026

San FranciscoRemoteFull-timeEngineering

Skills & Technologies

AWSGCPKubernetesPyTorch

Ready to apply?

Opens Physicalintelligence's career page

Apply Now

Back to jobs

Similar Jobs

ML Engineer (Infra)

Toss

ML Infra Engineer

Physicalintelligence

San Francisco

Staff Software Developer Ml Infrastructure Core Infra

Google

Principal AI/ML Engineer, AV ML Infra

General Motors

GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America +1 more

Principal AI and ML Infra Software Engineer, GPU Clusters

Nvidia

US, CA, Santa Clara

ML Infra/Systems Engineer

RFS Group

San Francisco, CA

Senior Software Engineer, ML Infra

Roblox

San Mateo, CA, United States +1 more

Senior Software Engineer, ML Infra

Verily

Mountain View - Mayfield, United States of America

ML Infra Engineer - Supercomputing

Physicalintelligence

San Francisco

ML Platform Engineer II, Feature & Embeddings Infra

Etsy

Brooklyn, NY, United States of America

Staff AI/ML Engineer - AV ML Infra

General Motors

GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America +1 more

Senior AI/ML Engineer - AV Infra

General Motors

GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America +1 more

Principal Staff AI/ML Engineer - AV ML Infra

General Motors

GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America +1 more

Senior AI/ML Fullstack Engineer - AV ML Infra

General Motors

GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America

Staff AI/ML Fullstack Engineer - AV ML Infra

General Motors

GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America

Staff Software Engineer, ML Infra & Distributed Systems

Tubitv

San Francisco, CA; Los Angeles, CA; New York, NY (Hybrid); USA - Remote +1 more

Software Engineer, ML Infra

Newsbreak

Mountain View, California, United States