Skills demand · updated 2026-09-28
JAX jobs at AI companies: 249 open roles name it
249 of 14,904 open job descriptions at 286 AI companies mention JAX (1.7%), across 67 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 | 40% |
| 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 67 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 249
Newest 30 shown.
-
ML Runtime and Kernel Engineer - Core ML
InfrastructureRemote US Canadatoday
-
ML Algorithm Mapping and Performance Engineer, Core ML
EngineeringRemote UStoday
-
2027 Summer Intern, PhD, Machine Learning Research
ResearchInternUStoday
-
2027 Summer Intern, MS/PhD, Perception, Machine Learning
ResearchInternUS3d
-
2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning
ResearchInternUS3d
-
Research Engineer - Auto Research
ResearchUS4d
-
Research Engineer - ML Infrastructure
ResearchUS4d
-
2027 Summer Intern, PhD, Software Engineer, Simulation
ResearchInternUS4d
-
2027 Summer Intern, MS/PhD, Road Understanding, ML Engineer
ResearchInternUS5d
-
SDET II
OtherSeniorUS$140k–170k5d
-
AI engineer
EngineeringSeniorRemote US$198k–259k6d
-
AI research scientist
ResearchSeniorRemote US$199k–325k6d
-
AI engineer (UK)
EngineeringSeniorRemote UK & Ireland6d
-
2027 Summer Intern, MS/PhD, Machine Learning Engineer
ResearchInternUS6d
-
Research Scientist, World Model Post-Training
ResearchUS$213k–263k6d
-
Staff Research Scientist, Reinforce Learning
ResearchStaff+UK & Ireland6d
-
2027 Summer Intern, MS/PhD, Software Engineer, Simulation Evaluation ML Model
ResearchInternUS7d
-
Research, Mid Training
ResearchRemote US$350k–475k7d
-
Research Intern, Frontier Agents (Summer 2027)
ResearchInternUS10d
-
Research Intern, Frontier Agents (Winter 2027)
ResearchInternUS10d
-
Research Intern, Inference (Summer 2027)
ResearchInternUS10d
-
Research Intern, Inference (Winter 2027)
ResearchInternUS10d
-
Research Intern, Model Shaping (Summer 2027)
ResearchInternUS Europe10d
-
Research Intern, Model Shaping (Winter 2027)
ResearchInternUS Europe10d
-
2027 Summer Intern, MS/PhD, Sim-Realism ML Infrastructure
ResearchInternUK & Ireland£65k–65k10d
-
Machine Learning Engineer, Perception
EngineeringUS$175k–215k10d
-
2027 Summer Intern, MS/PhD, Machine Learning, Simulation Realism
ResearchInternUK & Ireland£65k–65k11d
-
Staff Tech Lead Manager, Machine Learning, Simulator Evaluation
EngineeringLead / ManagerUS$251k–310k11d
-
Senior ML Engineer (AI Research, Physical AI)
ResearchSeniorRemote UK & Ireland Europe11d
-
2027 Summer Intern, PhD, Machine Learning, Computer Vision
ResearchInternUS12d
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 286 companies, plus the weekly index. Free.