🇺🇸JAX jobs in California
196 open roles mentioning JAX in California as of Nov 15, 2026. Waymo, xAI, Anthropic, Sciforium and Achira have the most. 3 are remote and the median advertised salary is $258K.
My jobs then shows how many are new each time you come back. No account, no email.
Skills asked for in JAX roles in California
Job titles in JAX roles in California
Bars show the share of these 196 roles; one role can count in several rows.
- Staff/Sr. Staff/Principal AI research scientistWriter · 🇺🇸San Francisco, CA$234K - $384KHybridLead and abovePythonMachine LearningLLM
- AI research scientistWriter · 🇺🇸San Francisco, CA$168K - $233KHybridPythonMachine LearningLLM
- Packaging Thermal and Mechanical Analysis Internship - Summer 2027Lightmatter · 🇺🇸Mountain View, CA—InternPythonMachine LearningPyTorch
- Member of Technical Staff - ML Infrastructure Engineer, RL Post-trainingPreference Model · 🇺🇸San Francisco$200K - $350KGCPLLMPyTorch
- Member of Technical Staff - Research Engineer, Post-trainingPreference Model · 🇺🇸San Francisco$200K - $350KPythonMachine LearningLLM
- Member of Technical Staff - Research Engineer, RL EnvironmentsPreference Model · 🇺🇸San Francisco$200K - $350KPythonMachine LearningLLM
- Sr. Model Engineer, RL DataCursor · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- Model Engineer, RL DataCursor · 🇺🇸Palo Alto, CA$160K - $250KRustPythonC++
- Member of Technical Staff, Pretraining DataMirendil · 🇺🇸San Francisco$300K - $400KPythonPyTorchLLM evaluation
- Sr. Model Engineer, VoicexAI · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- GPU Kernel EngineerSciforium · 🇺🇸San Francisco, CA$190K - $250KPythonC++LLM
- Senior Forward Deployed EngineerSuki · 🇺🇸Redwood City, CA$225K - $250KSeniorGoPythonTypeScript
- 2027 Summer Intern, PhD, Machine Learning - Generative World Models & Controllable SimulationWaymo · 🇺🇸Mountain View, California, United States—InternPythonMachine LearningPyTorch
- 2027 Summer Intern, Machine Learning, MS/PhD, Simulation Realism (World Model)Waymo · 🇺🇸Mountain View, California, United States—InternPythonMachine LearningPyTorch
- Applied ResearcherRiver AI Inc. · 🇺🇸Palo Alto, CA$200K - $420KMachine LearningPyTorchReinforcement learning
- Staff Machine Learning Engineer, AI InfrastructureWaymo · 🇺🇸Mountain View, CA, USA; San Francisco, CA, USA; New York…$251K - $310KLead and abovePythonC++Machine Learning
- Research Scientist, Discovery (Engineering & Applied Sciences)Anthropic · 🇺🇸San Francisco, CA$500K - $850KPythonMachine LearningLLM
- Sr. Model Engineer, RL Training FrameworkxAI · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- Senior Machine Learning Engineer (Growth)Discord · 🇺🇸San Francisco Bay Area$220K - $275KSeniorPythonMachine LearningPyTorch
- Senior Staff Machine Learning Engineer, Perception, Simulation RealismWaymo · 🇺🇸Mountain View, California, United States; San Francisco…$298K - $368KLead and abovePythonC++Machine Learning
- Quantum Characterization, Verification, and Validation EngineerPsiQuantum · 🇺🇸Milpitas, California$89K - $149KPythonPyTorchTensorFlow
- Machine Learning Engineer - New GradSambaNova · 🇺🇸Austin, Texas, United States; San Jose, California, Unite…$131K - $197KJuniorPythonMachine LearningLLM
- Research Engineer, InfrastructureHarvey · 🇺🇸San Francisco$230K - $450KHybridPythonPyTorchLLM evaluation
- Sr. Model Engineer, Post-Training & RLxAI · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- Sr. Model Engineer, Mid-TrainingxAI · 🇺🇸Palo Alto, CA$220K - $357KSeniorRustPythonC++
- Software Engineer Intern, Autonomy Software [Summer 2027]Zoox · 🇺🇸Foster City, CA—InternPythonC++Machine Learning
- Senior Robotics Engineer, Robot LearningSenra Systems · 🇺🇸Cypress, California, United States$200K - $260KSeniorPythonC++PyTorch
- Software Engineer - ManipulationApptronik · 🇺🇸Sunnyvale, CA$180K - $240KPythonC++PyTorch
- Autonomy Engineer Intern, Reinforcement Learning, 2027 (HQ)Skydio · 🇺🇸San Mateo, California, United States—InternPythonC++Machine Learning
- 2027 Summer Intern, MS/PhD, Machine Learning, Driver Refinement FoundationsWaymo · 🇺🇸San Francisco, CA, USA · +1 more location—InternPythonC++SQL
- 2027 Summer Intern, MS/PhD, Software Engineer, ML Frameworks and EfficiencyWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningPyTorch
- Senior Machine Learning Engineer, Scenes, Traffic ControlsWaymo · 🇺🇸Mountain View, California, United States, San Francisco…$213K - $263KSeniorPythonC++SQL
- 2027 Summer Intern, PhD, Research, Multimodal Foundation ModelsWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningPyTorch
- 2027 Summer Intern, PhD, AI Foundations - RecipesWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningPyTorch
- AI Performance Engineer - New GradSambaNova · 🇺🇸Austin, Texas, United States; San Jose, California, Unite…$131K - $197KJuniorPythonC++Machine Learning
- 2027 Summer Intern, PhD, Research, VLM Harnesses for AV PlanningWaymo · 🇺🇸Mountain View, CA, USA—InternMachine LearningLLMPyTorch
- 2027 Summer Intern, PhD, Machine Learning - Simulation & Closed-Loop TrainingWaymo · 🇺🇸Mountain View, California, United States—InternPythonC++Machine Learning
- Staff Engineer, Hivemind Reasoning (R6292)Shield AI · 🇺🇸San Mateo, CA, US—Lead and abovePyTorchTensorFlowReinforcement learning
- Senior Manager, Engineering - Hivemind Reasoning (R6291)Shield AI · 🇺🇸San Mateo, CA, US—SeniorPyTorchTensorFlowReinforcement learning
- 2027 Summer Intern, PhD, Research Scientist, Visual ReasoningWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningLLM
- 2027 Summer Intern, PhD, Research, Behavior Post-trainingWaymo · 🇺🇸Mountain View, CA, USA; San Francisco, CA, USA—InternMachine LearningLLMPyTorch
- Software Engineer, ML Training OptimizationZoox · 🇺🇸Foster City, CA$177K - $270KPythonC++PyTorch
- Senior/Staff Software Engineer, ML Training OptimizationZoox · 🇺🇸Foster City, CA$254K - $350KLead and abovePythonC++PyTorch
- Senior Staff Software Engineer, StorageCrusoe · 🇺🇸Sunnyvale, CA - US$245K - $290KLead and aboveRustGoC++
- 2027 Summer Intern, MS/PhD, SemanticsWaymo · 🇺🇸Mountain View, California, USA—InternPythonMachine LearningPyTorch
- Senior Perception Learning EngineerApptronik · 🇺🇸Sunnyvale, CA$220K - $250KSeniorPythonC++PyTorch
- 2027 Summer Intern, MS/PhD, Machine Learning, Planning/PredictionWaymo · 🇺🇸Mountain View, CA, USA—InternPythonC++Machine Learning
- 2027 Summer Intern, MS/PhD, Perception, Model LifecycleWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningPyTorch
- 2027 Summer Intern, PhD, Research, AV PlanningWaymo · 🇺🇸Mountain View, CA, USA—InternPythonMachine LearningLLM
- 2027 Summer Intern, PhD, Research, Post TrainingWaymo · 🇺🇸Mountain View, CA, USA: San Francisco, CA, USA—InternMachine LearningPyTorchReinforcement learning
Showing the newest 50. See all 196 in search
Who is hiring for JAX roles in California
What JAX roles in California pay, by level
| Level in the title | Median advertised salary | Roles counted |
|---|---|---|
| No level in the title | $250K | 69 |
| Senior | $238K | 31 |
| Lead and above | $280K | 35 |
Medians of the midpoints of advertised yearly US-dollar ranges; a level needs at least 15 such roles. Advertised, not verified pay.
Questions about JAX jobs in California
- How many JAX jobs in California are open right now?
- There are 196 open JAX jobs in California at 60 companies on Jobs Every Hour as of Nov 15, 2026. 9 of them were posted in the last seven days. The list is rebuilt every hour from the employers' own careers pages.
- Which companies have the most JAX jobs in California?
- Right now the most openings are at Waymo (64), xAI (13), Anthropic (9), Sciforium (7) and Achira (6). The ranking changes as companies open and close roles.
- Are there remote JAX jobs in California?
- 3 of the 196 open JAX jobs in California (2%) are listed as remote. Remote roles are often limited to certain countries or time zones, so check the location line of each job.
- What do JAX jobs in California pay?
- The median advertised salary is $258K, based on 138 open roles that publish a yearly US-dollar range. The middle half of those roles advertise between $212K and $300K. These are advertised ranges, not verified pay. Most employers here are technology companies, so the figures run above the wider market for the same work.
- Is it hard to get one of the JAX jobs in California?
- 43 of the 196 open JAX jobs in California (22%) are for beginners: junior roles and internships. Of the 80 postings that state experience, the median asks for 5 years. 60 companies are hiring and 9 roles were posted in the last seven days. Matching your resume to the skills each posting names helps most; the free resume builder shows the match.
- How is AI changing JAX jobs in California?
- 190 of the 196 open JAX jobs in California (97%) ask for AI or machine-learning skills, most often PyTorch, Machine Learning, AI. Employers still list 196 such roles today at 60 companies. Counted from the postings' own text on Nov 15, 2026.
- What skills do JAX jobs in California ask for?
- Most often JAX (100%), PyTorch (94%), Python (77%), Machine Learning (67%), Deep learning (46%) of the 196 open postings, counted from their own text on Nov 15, 2026.
- Which job titles come up most in JAX jobs in California?
- 2027 Summer Intern (30), Research Engineer (13), Member of Technical Staff (10), Senior Machine Learning Engineer (8), Machine Learning Engineer (8): the most common titles among the 196 open roles, without the level and place notes in brackets.
- Are JAX jobs in California in demand in 2026?
- 196 JAX jobs in California are open on Jobs Every Hour today at 60 companies, 0.1% of all 148,162 open jobs here; 9 were posted in the last seven days. The median advertised salary is $258K. 3 (2%) can be done remotely.
- How do I apply for JAX jobs in California?
- Open a job and press “Apply on company site”. You go straight to the employer's own careers page and apply there. Jobs Every Hour needs no account and charges nothing for applying.
More like JAX jobs in California
- Jobs in California 28,322
- JAX jobs 375
Next to it
- Python jobs in California 7,177
- LLM jobs in California 3,007
- C++ jobs in California 2,665
- AWS jobs in California 2,475
- Machine Learning jobs in California 2,435
- SQL jobs in California 2,406
- Computer Vision jobs in California 2,106
- Go jobs in California 2,029
- Kubernetes jobs in California 2,003
- Security clearance jobs in California 1,895
- Linux jobs in California 1,774
- Sensor fusion jobs in California 1,734




















