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.
8 AI roles requiring this skill.
| Company | Title | Sector | AI score | Stage |
|---|---|---|---|---|
| NVIDIA | Senior Research Scientist, Multimodal Foundation Models and Robotics | Semiconductors | 10 | L2 |
| NVIDIA | Research Scientist, Generalist Embodied Agent Research - PhD New College Grad 2026 | Semiconductors | 10 | L4 |
| NVIDIA | Senior Research Scientist, Multimodal Foundation Models and Robotics | Semiconductors | 10 | L2 |
| Intel | Neuromorphic Applications Researcher- Temporary Position | Semiconductors | 9 | L6 |
| Intel | AI Algorithm Research Intern – Neuromorphic Computing | Semiconductors | 9 | L0 |
| NVIDIA | Senior Robotics Research Engineer, Robotics and AI for Drug Discovery | Semiconductors | 9 | L4 |
| NVIDIA | Research Scientist, Robotics Research - PhD New College Grad 2026 | Semiconductors | 9 | L4 |
| NVIDIA | Senior Robotics Research Scientist | Semiconductors | 9 | L6 |