Skills demand · updated 2026-09-29
JAX jobs at AI companies: 252 open roles name it
252 of 14,934 open job descriptions at 282 AI companies mention JAX (1.7%), across 69 companies. From the companies' own careers feeds, refreshed every 6 hours.
Companies asking for JAX most
| Company | Roles | Share of their roles |
|---|---|---|
| Waymo | 49 | 14% |
| Thinking Machines Lab | 19 | 39% |
| Lila Sciences | 13 | 12% |
| Cohere | 13 | 9% |
| Anthropic | 10 | 2% |
| Graphcore | 9 | 5% |
| xAI | 7 | 2% |
| Waabi | 7 | 8% |
| Together AI | 6 | 8% |
| Isomorphic Labs | 5 | 16% |
| Nuro | 5 | 5% |
| Wayve | 4 | 2% |
| Nebius | 4 | 1% |
| Extropic | 4 | 40% |
| CoreWeave | 4 | 1% |
All 69 companies naming JAX, and the same for every tracked skill, as a table and CSV in AI Hiring Index Pro. Vendors: this is where your category is being hired for.
Open roles that name JAX 252
Newest 30 shown.
-
ML Runtime and Kernel Engineer - Core ML
InfrastructureUS Canada1d
-
ML Algorithm Mapping and Performance Engineer, Core ML
EngineeringUS1d
-
2027 Summer Intern, PhD, Machine Learning Research, Planning/Prediction
ResearchInternUS1d
-
2027 Summer Intern, MS/PhD, Perception, Machine Learning
ResearchInternUS4d
-
2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning
ResearchInternUS4d
-
Research Engineer - Auto Research
ResearchUS5d
-
Research Engineer - ML Infrastructure
ResearchUS5d
-
2027 Summer Intern, PhD, Software Engineer, Simulation
ResearchInternUS5d
-
2027 Summer Intern, MS/PhD, Road Understanding, ML Engineer
ResearchInternUS6d
-
SDET II
EngineeringUS$140k–170k6d
-
AI engineer
EngineeringUS$220k–259k7d
-
AI research scientist
ResearchUS$199k–325k7d
-
AI engineer (UK)
EngineeringUK & Ireland7d
-
2027 Summer Intern, MS/PhD, Machine Learning Engineer
ResearchInternUS7d
-
Research Scientist, World Model Post-Training
ResearchUS$213k–263k7d
-
Staff Research Scientist, Reinforce Learning
ResearchStaff+UK & Ireland7d
-
2027 Summer Intern, MS/PhD, Software Engineer, Simulation Evaluation ML Model
ResearchInternUS8d
-
Research, Mid Training
ResearchUS$350k–475k8d
-
Research Intern, Frontier Agents (Summer 2027)
ResearchInternUS11d
-
Research Intern, Frontier Agents (Winter 2027)
ResearchInternUS11d
-
Research Intern, Inference (Summer 2027)
ResearchInternUS11d
-
Research Intern, Inference (Winter 2027)
ResearchInternUS11d
-
Research Intern, Model Shaping (Summer 2027)
ResearchInternUS Europe11d
-
Research Intern, Model Shaping (Winter 2027)
ResearchInternUS Europe11d
-
2027 Summer Intern, MS/PhD, Sim-Realism ML Infrastructure
ResearchInternUK & Ireland£65k–65k11d
-
Machine Learning Engineer, Perception
EngineeringUS$175k–215k11d
-
2027 Summer Intern, MS/PhD, Machine Learning, Simulation Realism
ResearchInternUK & Ireland£65k–65k12d
-
Staff Tech Lead Manager, Machine Learning, Simulator Evaluation
EngineeringLead / ManagerUS$251k–310k12d
-
Senior ML Engineer (AI Research, Physical AI)
ResearchSeniorRemote UK & Ireland Europe12d
-
2027 Summer Intern, PhD, Machine Learning, Computer Vision
ResearchInternUS13d
See also: all skills in demand · posted salary ranges · Machine Learning Engineer jobs · Research Scientist jobs · Research Engineer jobs.
New research roles at AI companies, every Monday. The week's openings in this function across 277 companies, plus the weekly index. Free.