AI Jobs
AI Jobs
Post a job
Sign up Sign in

Full Time PyTorch AI Jobs in California

Browse 37 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 Full Time PyTorch AI Jobs in California
Share:
AI Jobs
PyTorch Jobs in AI
Full Time 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
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
TML
Research Engineer, Infrastructure, Numerics
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs Distributed Systems jobs
$350K – $475K 12D
TML
Research, Vision Expertise
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 12D
TML
Research Engineer, Infrastructure, RL Systems
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs RLHF jobs
$350K – $475K 12D
TML
Software Engineer, Supercomputing
Thinking Machines Lab / San Francisco, United States / Hybrid
Python jobs PyTorch jobs TensorFlow jobs
$350K – $475K 12D
TML
Research Engineer, Infrastructure, Kernels
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs CUDA jobs
$350K – $475K 12D
TML
Research, Post-Training Data
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs TensorFlow jobs JAX jobs
$350K – $475K 12D
TML
Research Engineer, Infrastructure, Training Systems
Thinking Machines Lab / San Francisco, United States / Hybrid
PyTorch jobs JAX jobs Megatron jobs
$350K – $475K 12D
PA
Staff Machine Learning Engineer
Primer AI / Washington, DC ; San Francisco, CA / Remote / Staff
Python jobs PyTorch jobs vLLM jobs
13D
O
Researcher, Frontier Cybersecurity Risks
OpenAI / San Francisco, United States
TensorFlow jobs Excel jobs
13D
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
A
Computer Vision Engineer
Archer / San Jose, United States
Python jobs PyTorch jobs CUDA jobs
$140K – $260K 16D
LA
Senior Accountant - Accounts Payable & Procure-to-Pay
Lightning AI / San Francisco, United States / Senior
Go jobs NetSuite jobs
$100K – $150K 18D
XM
Senior Reinforcement Learning Engineer - Robotics
XPeng Motors / Santa Clara, United States / Senior
Python jobs PyTorch jobs C++ jobs
$175K – $296K 18D
X
ML Infrastructure Engineer
xAI / Palo Alto, United States
Python jobs PyTorch jobs JAX jobs
$180K – $440K 26D

Frequently asked questions

What is the average salary for Full Time PyTorch jobs?

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

Which companies are hiring for Full Time PyTorch jobs?

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

Are there remote Full Time PyTorch jobs?

Yes, 4 of 37 are remote.