AI Jobs
AI Jobs
Post a job
Sign up Sign in

PyTorch AI Jobs in United States

Browse 62 AI jobs in United States for Aug 2026 with roles open at xAI, Graphcore, and Torc Robotics paying $24K to $500K/year. Work as a ML Infrastructure Engineer, Principal Engineer, and Staff Engineer.

PyTorch is required in 18% of ML Engineer postings — the most common role requiring this skill.

Receive emails of PyTorch AI Jobs in United States
Share:
AI Jobs
PyTorch Jobs in AI
PyTorch AI Jobs in United States
AI Deployment Lead jobs Art Director jobs Business Development Manager jobs Chief of Staff jobs DevOps jobs Docker jobs Field Engineer jobs Go jobs Grafana jobs GraphQL jobs Helm jobs Hugging Face jobs Illustrator jobs Jetpack Compose jobs Jira jobs LoRA jobs Marketing Lead jobs Pandas jobs PyTorch jobs Recruiter jobs Recruiting Coordinator jobs Redis jobs Research Engineer jobs Sales Manager jobs Salesforce jobs Solutions Architect jobs
Filter by type
Remote Full Time Internship
TML
Research Engineer, Infrastructure, RL Systems
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs RLHF jobs
$350K – $475K 13D
TML
Software Engineer, Supercomputing
Thinking Machines Lab / San Francisco, United States / Hybrid
Python jobs PyTorch jobs TensorFlow jobs
$350K – $475K 13D
TML
Research Engineer, Infrastructure, Kernels
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs CUDA jobs
$350K – $475K 13D
TML
Research, Post-Training Data
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 13D
TML
Research Engineer, Infrastructure, Training Systems
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs Megatron jobs
$350K – $475K 13D
PA
Staff Machine Learning Engineer
Primer AI / Washington, DC ; San Francisco, CA / Remote / Staff
Python jobs PyTorch jobs vLLM jobs
13D
MM
Lead Machine Learning Engineer - Localization
May Mobility / United States / Remote / Senior
Python jobs PyTorch jobs TensorFlow jobs
$235K – $285K 13D
TR
Staff, ML Engineer - Scene Generation
Torc Robotics / Ann Arbor, United States / Remote / Staff
Python jobs PyTorch jobs CUDA jobs
$250K – $300K 13D
O
Researcher, Frontier Cybersecurity Risks
OpenAI / San Francisco, United States
TensorFlow jobs Excel jobs
14D
QC
Internship – Data Management
QuEra Computing / Boston, United States
PyTorch jobs Pandas jobs
$30/hr – $40/hr 14D
D
Solutions Architect - AI Natives Business, Strategic
Databricks / California ; Washington, DC / Remote / Staff
Python jobs PyTorch jobs TensorFlow jobs
$180K – $248K 14D
D
Solutions Architect - Digital Native Business, Named Accounts
Databricks / California ; Washington, DC / Remote / Staff
Python jobs PyTorch jobs TensorFlow jobs
$180K – $248K 14D
P
AI Engineer, Internship - Summer 2026 - Applications Open Now
Postman / San Francisco, CA ; Berkeley, CA
Python jobs PyTorch jobs LoRA jobs
16D
A
Computer Vision Engineer
Archer / San Jose, United States
Python jobs PyTorch jobs CUDA jobs
$140K – $260K 17D
LA
Senior Accountant - Accounts Payable & Procure-to-Pay
Lightning AI / San Francisco, United States / Senior
Go jobs NetSuite jobs
$100K – $150K 18D
LA
Director of Customer Experience
Lightning AI / New York, United States
Kubernetes jobs Slurm jobs
$155K – $220K 19D
XM
Senior Reinforcement Learning Engineer - Robotics
XPeng Motors / Santa Clara, United States / Senior
Python jobs PyTorch jobs C++ jobs
$175K – $296K 19D
LA
Director of Sales
Lightning AI / New York, United States
Cadence jobs
$450K – $500K 19D
LA
Director of Strategy and Operations
Lightning AI / New York, United States
Go jobs
$180K – $250K 19D
LA
VP of Solution Engineering
Lightning AI / New York, United States
Cadence jobs
$250K – $400K 19D

Frequently asked questions

What is the average salary for PyTorch jobs?

62 active listings have salaries ranging from $24K to $500K/year.

Which companies are hiring for PyTorch jobs?

Top companies hiring include xAI, Graphcore, and Torc Robotics.

Are there remote PyTorch jobs?

Yes, 11 of 62 are remote.