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.
42 AI roles requiring this skill.
| Company | Title | Sector | AI score | Stage |
|---|---|---|---|---|
| Cohere | Senior Member of Technical Staff, Multimodal AI | AI Frontier | 9 | L2 |
| Cohere | Member of Technical Staff, Modeling | AI Frontier | 9 | L3 |
| Lila Sciences | Co-op, LLMs for Decision Making | AI Frontier | 8 | L4 |
| Physical Intelligence | ML Infra Engineer (TPU/Jax/Optimization) | AI Frontier | 8 | L0 |
| Cohere | Staff Software Engineer, GPU Infrastructure (HPC) | AI Frontier | 8 | L3 |
| Cohere | Member of Technical Staff, Data Analysis and Evaluation | AI Frontier | 8 | L0 |
| Cohere | Member of Technical Staff, MLE (UK/EU) | AI Frontier | 8 | L4 |
| Cohere | Member of Technical Staff, Integration/RL Team (Research Engineer) | AI Frontier | 8 | L2 |
| Cohere | Member of Technical Staff, MLE (Korea) | AI Frontier | 8 | L4 |
| Cohere | Senior Member of Technical Staff, MLE (Middle East) | AI Frontier | 8 | L4 |
| Physical Intelligence | ML Infra Engineer | AI Frontier | 8 | L0 |
| Lila Sciences | Co-op, Machine Learning for Digital Twins | AI Frontier | 7 | L2 |