242 active AI roles across 39 companies mention JAX. Category: Frameworks & Tools.
New AI postings mentioning JAX per week — 138 total over 12 weeks.
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.
4 AI roles requiring this skill.
| Company | Title | Sector | AI score | Stage |
|---|---|---|---|---|
| Salesforce | Research Scientist - Salesforce AI Research | Enterprise | 9 | L2 |
| Figma | AI Applied Scientist | Enterprise | 9 | L2 |
| Adobe | Principal Engineer AI/ML | Enterprise | 8 | L6 |
| UiPath | Senior Principal Applied Scientist | Enterprise | 7 | L2 |