AI Jobs
AI Jobs
Post a job
Sign up Sign in

PyTorch AI Jobs in California

Browse 36 AI jobs in California for Aug 2026 with roles open at xAI, Graphcore, and Lightning AI paying $100K to $475K/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 California
Share:
AI Jobs
PyTorch Jobs in AI
PyTorch AI Jobs in California
BGP jobs Business Development Manager jobs Chief of Staff jobs Data Analyst jobs DevOps jobs Docker jobs Field Engineer jobs Go jobs Grafana jobs GraphQL jobs Helm jobs Hugging Face 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 XLA jobs
Filter by type
Remote Full Time Internship
S
Senior Principal Compiler Engineer
SambaNova / Austin, TX ; San Jose, CA / Remote / Staff
PyTorch jobs TensorFlow jobs
$180K – $240K 2D
O
Machine Learning Engineer, API Multicloud
OpenAI / San Francisco, United States
Python jobs PyTorch jobs TensorFlow jobs
4D
C
Senior Staff Software Engineer, DC Infrastructure
Crusoe / San Francisco, CA ; Sunnyvale, CA / Staff
Python jobs PyTorch jobs Kubernetes jobs
$250K – $300K 4D
O
Machine Learning Data Scientist, Forecasting
OpenAI / San Francisco, United States / Hybrid
Python jobs PyTorch jobs TensorFlow jobs
4D
G
Principal Engineer
Graphcore / Austin, TX ; Milpitas, CA / Staff
PyTorch jobs Kubernetes jobs Terraform jobs
5D
G
Staff Engineer
Graphcore / Austin, TX ; Milpitas, CA / Staff
PyTorch jobs Kubernetes jobs Terraform jobs
5D
SSI
Staff Software Engineer AI/ML
Samsung Semiconductor, Inc. / San Jose, United States / Staff
Python jobs PyTorch jobs TensorFlow jobs
$163K – $253K 6D
N
Senior Machine Learning Engineer, Recommendation & AI Applications
NewsBreak / Firestone Park, United States / Senior
Python jobs PyTorch jobs TensorFlow jobs
$195K – $230K 9D
LA
Senior Software Engineer, Training & Experimentation
Lightning AI / London, GB ; San Francisco, CA / Senior
Python jobs PyTorch jobs Kubernetes jobs
$180K – $250K 9D
LA
Senior Software Engineer, Core Platform
Lightning AI / London, GB ; San Francisco, CA / Senior
Python jobs PyTorch jobs Kubernetes jobs
$180K – $250K 9D
LA
Senior Software Engineer, Agents
Lightning AI / London, GB ; San Francisco, CA / Senior
Python jobs PyTorch jobs Kubernetes jobs
$180K – $250K 9D
C
Research Engineer, Mid-Training
Cognition / San Francisco, United States
Python jobs PyTorch jobs
10D
C
Research Engineer, ML Infrastructure
Cognition / San Francisco, United States
Python jobs PyTorch jobs C++ jobs
10D
LA
Senior Accountant - General Ledger
Lightning AI / San Francisco, United States / Senior
Cadence jobs Excel jobs NetSuite jobs
$100K – $150K 10D
TML
Research, Post-Training
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 12D
TML
HR Business Partner
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs
$190K – $300K 12D
TML
Research Engineer, Infrastructure, Inference
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs vLLM jobs
$350K – $475K 12D
TML
Research, Pre-Training Data
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 12D
TML
Research, Pre-Training Science
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 12D
TML
Research, Audio Expertise
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 12D

Frequently asked questions

What is the average salary for PyTorch jobs?

36 active listings have salaries ranging from $100K to $475K/year.

Which companies are hiring for PyTorch jobs?

Top companies hiring include xAI, Graphcore, and Lightning AI.

Are there remote PyTorch jobs?

Yes, 4 of 36 are remote.