Latest AI JobsBrowse Jobs

PyTorch Jobs in CA

404 open roles, updated daily

Filters2
404 jobs foundClear all
Luma AI logo
Research Scientist / Engineer – Training Infrastructure
Luma AI
12d ago

Apply for the Training Infrastructure Research Scientist role at Luma AI in Redwood City, CA. Build distributed systems for multimodal models. $195K-$395K.

Redwood City, CAResearch & AIFull-timeSeniorRemoteOpen · checked 8d ago
PyTorchCUDADistributed Systems
$195K – $395KView Job
H
Inference Optimization Engineer
hedra
12d ago

Apply for the Inference Optimization Engineer role at hedra in San Francisco. Work on visual intelligence models, inference, and GPU systems.

San FranciscoResearchFull-timeOpen · checked 8d ago
PythonPyTorchC++
Salary not listedView Job
DoorDash USA logo
Software Engineer, Machine Learning Platform - MDX
DoorDash USA
12d ago

Apply for the Software Engineer, Machine Learning Platform role at DoorDash in San Francisco, Sunnyvale, or Seattle. Base salary $130,600 — $192,000 USD.

San Francisco, CA; Sunnyvale, CA; • Seattle, WA313 Infrastructure EngineeringFull-timeMid · 2+ yrsOpen · checked 8d ago
PythonPyTorchTensorFlow
$130,600 — $192K USDView Job
Fireworks AI logo
Member of Technical Staff, Growth
Fireworks AI
12d ago

Apply for the Member of Technical Staff, Growth role at Fireworks AI in San Mateo. Build growth platforms and developer experiences for enterprise AI.

San MateoEngineeringFull-timeSenior · 5+ yrsRemoteOpen · checked 8d ago
PyTorchLLMSQL
$200K – $290KView Job
S
GPU Kernel Engineer
Sciforium
12d ago

Apply for the GPU Kernel Engineer role at Sciforium in San Francisco, CA. Salary: $190K–$250K with equity. Work on custom GPU kernels for AI systems.

San Francisco, CAEngineeringFull-timeSenior · 5+ yrsOpen · checked 8d ago
PythonPyTorchJAX
$190K – $250KView Job
DatologyAI logo
Research Scientist, Post-Training
DatologyAI
12d ago

Apply for the Research Scientist, Post-Training role at DatologyAI in San Mateo. Salary $180,000 - $300,000. Lead foundation model data curation research.

San MateoResearchFull-timeSenior · 3+ yrsOpen · checked 8d ago
PyTorchSparkDeep Learning
$180K to $300K.View Job
Niantic Spatial logo
AI Engineer, Computer Vision
Niantic Spatial
12d ago

Apply for the AI Engineer, Computer Vision role at Niantic Spatial in San Francisco, CA. Hybrid schedule with salary up to $221K.

San Francisco, CAEngineeringFull-timeMidRemoteOpen · checked 8d ago
PythonPyTorchComputer Vision
$165.6K – $221KView Job
Thinking Machines logo
RL Systems - Infra
Thinking Machines
12d ago

Apply for the RL Systems Infra role at Thinking Machines in San Francisco. Distributed training, RL workloads, $425K-$475K salary, visa support.

San FranciscoResearch Infrastructure (ML Infrastructure and Training Stack)Full-timeSeniorOpen · checked 8d ago
PyTorchJAXKubernetes
$425K - $475K USDView Job
Lila Sciences logo
Principal Machine Learning Engineer, Applied AI
Lila Sciences
13d ago

Apply for the Principal Machine Learning Engineer role at Lila Sciences in Cambridge, MA or San Francisco, CA. Salary: $252k-$336k USD.

Cambridge, MA USA; San Francisco, CA USAAIFull-timeSenior · 2+ yrsOpen · checked 8d ago
PythonPyTorchMachine Learning
$252K — $336K USDView Job
Lila Sciences logo
Senior / Staff Machine Learning Engineer, Applied AI
Lila Sciences
13d ago

Join Lila Sciences in Cambridge or San Francisco as a Senior / Staff ML Engineer, Applied AI. Work on frontier models and scientific workflows.

Cambridge, MA USA; San Francisco, CA USAAIFull-timeSenior · 1+ yrsOpen · checked 8d ago
PythonPyTorchMachine Learning
$180K — $298K USDView Job
Everpure logo
Consulting Field Solutions Architect - AI/Analytics - Central Enterprise
Everpure
13d ago

Apply for the Consulting Field Solutions Architect role at Everpure in Chicago, Detroit, and more. Base salary $171,500 - $257,600 USD.

Chicago, Illinois; Detroit, Michigan; Indianapolis, Indiana; Kansas City, Kansas; Minneapolis, Minnesota; St. Louis, MissouriSalesFull-timeSenior · 8+ yrsOpen · checked 8d ago
PyTorchTensorFlowAWS
$171,500 — $257,600 USDView Job
W
2027 Summer Intern, MS/PhD, Research, Perception Foundation Models
Waymo
13d ago

Apply for the 2027 Summer Intern, Perception Foundation Models role at Waymo in Mountain View, CA. Hybrid onsite graduate machine learning internship.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 8d ago
PythonPyTorchTensorFlow
$70 — $70 USDView Job
B
Engineering Manager - Inference Performance
Baseten
13d ago

Apply for the Engineering Manager - Inference Performance role at Baseten in San Francisco. Lead GPU optimization teams and scale AI infrastructure.

San FranciscoEPDFull-timeSeniorRemoteOpen · checked 8d ago
PyTorchLLMSpark
$240K – $270KView Job
T
Member of Technical Staff - ML Infra
trajectory
13d ago

Build ML infrastructure for AI systems at Trajectory in San Francisco. Focus on training, inference, and GPU kernels. Apply on the official careers page.

San FranciscoMember of Technical StaffFull-timeSeniorOpen · checked 8d ago
PythonPyTorchJAX
Salary not listedView Job
Faire logo
Applied AI/ML Scientist, Intern
Faire
13d ago

Apply for the Applied AI/ML Scientist Intern role at Faire in San Francisco. Winter 2027 paid internship paying $75/hour for graduate students.

San Francisco, CAAlgorithms & DataInternshipInternshipOpen · checked 8d ago
PythonPyTorchSQL
Salary not listedView Job
Lyft logo
PhD Machine Learning Software Engineer Intern (Summer 2027)
Lyft
13d ago

Apply for the PhD Machine Learning Software Engineer Intern role at Lyft in San Francisco, CA. Summer 2027 hybrid internship focusing on applied AI research.

San Francisco, CAEarly TalentInternshipInternOpen · checked 8d ago
PythonPyTorchTensorFlow
$65-$68View Job
Niantic Spatial logo
Software Engineering Intern (Summer 2027)
Niantic Spatial
14d ago

Join Niantic Spatial in San Francisco for a Summer 2027 Software Engineering Internship. Earn $48-$70/hr working on physical AI and spatial intelligence.

San Francisco, CAEngineeringInternshipInternOpen · checked 8d ago
PythonPyTorchComputer Vision
$48 – $70 per hourView Job
applied logo
Senior Machine Learning Engineer - 3D Reconstruction
applied
14d ago

Apply for the Senior Machine Learning Engineer role at Applied Intuition in Sunnyvale. Focus on 3D reconstruction and neural simulation. Salary $190K-$260K.

SunnyvaleDana for Physical AIFull-timeSenior · 5+ yrsOpen · checked 8d ago
PythonPyTorchComputer Vision
$190K – $260KView Job
applied logo
Senior Machine Learning Engineer - Generative Models
applied
14d ago

Apply for the Senior Machine Learning Engineer role at applied in Sunnyvale. Work on generative models and neural simulation. Salary: $185K–$260K.

SunnyvaleDana for Physical AIFull-timeSenior · 5+ yrsOpen · checked 8d ago
PythonPyTorchComputer Vision
$185K – $260KView Job
Snorkel AI logo
Staff Applied AI Engineer - Enterprise AI Solutions
Snorkel AI
14d ago

Join Snorkel AI as a Staff Applied AI Engineer in Hybrid or Remote locations. Build enterprise AI solutions with a salary up to $300K plus equity.

New York City, NY (Hybrid); San Francisco, CA (Hybrid); United States (Remote)220 - Solutions PSFull-timeSenior · 3+ yrsRemoteOpen · checked 8d ago
PythonPyTorchLLM
$230K - $360KView Job
F
Senior Machine Learning Scientist
Flagship Pioneering, Inc.
14d ago

Apply as a Senior Machine Learning Scientist at Flagship Pioneering in Cambridge, UK. Build models and AI agents for somatic genomics.

Cambridge, UKArtificial IntelligenceFull-timeSeniorOpen · checked 8d ago
PythonPyTorchJAX
Salary not listedView Job
S
Senior HPC & GPU Infrastructure Engineer
Sciforium
14d ago

Apply for the Senior HPC & GPU Infrastructure Engineer role at Sciforium in San Francisco, CA. Manage high-density GPU clusters with a $150K-$220K salary.

San Francisco, CAEngineeringFull-timeSenior · 5+ yrsOpen · checked 8d ago
PythonPyTorchJAX
$150K – $220KView Job
B
Software Engineer - Inference Performance
Baseten
14d ago

Apply for the Software Engineer- Inference Performance role at Baseten in San Francisco. Salary $180K-$360K plus equity.

San FranciscoEPDFull-timeSeniorRemoteOpen · checked 8d ago
PythonPyTorchLLM
$180K – $360KView Job
wayve logo
Staff / Senior Machine Learning Engineer, Reinforcement Learning
wayve
14d ago

Apply for the Staff / Senior Machine Learning Engineer, RL role at Wayve in Sunnyvale, CA. Hybrid work, reinforcement learning, and competitive benefits.

Sunnyvale, California USAAV EngineeringFull-timeSeniorRemoteOpen · checked 8d ago
PythonPyTorchC++
Salary not listedView Job