New AI postings mentioning JAX per week — 138 total over 12 weeks.
242 active AI roles across 39 companies mention JAX. Category: Frameworks & Tools.
JAX is a skill in the "Frameworks & Tools" category. It currently appears in 242 active AI roles across 39 companies in our index.
The top employers with active AI roles mentioning JAX are: NVIDIA (51), Amazon (38), Google (34), Apple (30), Cohere (15).
Over the last 12 weeks, 138 new AI postings mentioned JAX. Demand is rising — up 139% in the last four weeks compared to the earliest four weeks in the window.
Roles requiring JAX are concentrated in: serving infrastructure (40%), post-training (17%), agents (14%). These stages follow a seven-stage AI lifecycle from data preparation through to shipped product.
Job postings that mention JAX most often also require: PyTorch, Python, Machine Learning, TensorFlow, Computer Architecture.
7 AI roles requiring this skill.
| Company | Title | Sector | AI score | Stage |
|---|---|---|---|---|
| Apptronik | Senior Reinforcement Learning Engineer | Robotics | 10 | L6 |
| Wayve | Applied Scientist, Wayve Labs | Robotics | 9 | L4 |
| Nuro | Software Engineer, ML Infrastructure, Optimization | Robotics | 8 | L3 |
| Apptronik | Senior Learning Perception Engineer - SLAM | Robotics | 8 | L4 |
| Apptronik | Senior Perception Learning Engineer | Robotics | 8 | L3 |
| Nuro | Senior/Staff Engineer, Machine Learning - Online Mapping | Robotics | 8 | L4 |
| Nuro | Senior/Staff Software Engineer, ML Infrastructure, Optimization | Robotics | 7 | L3 |