ML Infra Engineer (TPU/Jax/Optimization)

Physicalintelligence

Jan 23, 2026

Apply

Location: San Francisco
Workplace: Remote
Type: Full-time
Department: Engineering
Source: Ashby

Skills

AWSGCPKubernetesPyTorch

Similar Jobs

ML Infra Engineer

Physicalintelligence · San Francisco · Onsite

1+ year ago

ML Infra Engineer (Data Systems)

Physicalintelligence · San Francisco · Onsite

1 week ago

Staff Engineer - ML Infra / MLOps

Quince · Palo Alto, California, United States

$218K – $285

1 week ago

Software Engineer, ML Infra (Junior & New Grad)

Newsbreak · Mountain View, California, United States

$125K – $175K

9 months ago

Staff Software Engineer, ML Infra & Distributed Systems

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

$227K – $325K

10 months ago

Software Engineer, ML Infra

Newsbreak · Mountain View, California, United States

$180K – $225K

1+ year ago

Software Development Engineer I, ML Infra Services, Annapurna Labs

Amazon

2 weeks ago

Principal AI/ML Engineer, AV ML Infra

General Motors · GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America +1 · Remote

$276K – $341K

2 months ago

ML Infra/Systems Engineer

RFS Group · San Francisco, CA

4 months ago

Staff AI/ML Engineer - AV ML Infra

General Motors · GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America +1 · Remote

$219K – $335K

6 months ago

Senior AI/ML Engineer - AV Infra

General Motors · GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America +1 · Hybrid

$171K – $261K

6 months ago

Principal Staff AI/ML Engineer - AV ML Infra

General Motors · GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America +1 · Remote

$296K – $454K

6 months ago

Senior AI/ML Fullstack Engineer - AV ML Infra

General Motors · GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America · Remote

$171K – $261K

6 months ago

Staff AI/ML Fullstack Engineer - AV ML Infra

General Motors · GM Automation - Sunnyvale - GM Automation - Sunnyvale, United States of America · Remote

$219K – $335K

6 months ago

Senior Software Engineer, ML Evaluation Infra and Efficiency

Waymo · Remote

11 months ago

Founding Engineer - Software (Data/ML Infra)

RFS Group · Remote, Onsite

$180K – $220K

1 month ago