Companies in the United States using JAX
236 companies headquartered in the United States list JAX as a requirement in 554 open roles. Most of those roles are in AI and machine learning, backend engineering and data science teams. JAX most often appears next to Python, PyTorch, Machine Learning, CI/CD and Kubernetes. Roles that need it and state the pay sit at $212k a year in the middle.
Companies with the most open JAX roles
25 of 236 · the full list is in the API| Company | Size | Open roles | Teams hiring | Median stated pay |
|---|---|---|---|---|
 NVIDIA | 5000+ | 52 | AI and machine learning, Backend engineering, DevOps and infrastructure | $236k |
 TikTok | 1001-5000 | 30 | AI and machine learning, Data science, Engineering leadership | — |
 Google | 11-50 | 18 | AI and machine learning, Backend engineering, Solutions engineering | — |
 Thinking Machines Lab | 501-1000 | 16 | AI and machine learning, Web3 | $413k |
 Amazon | 5000+ | 14 | AI and machine learning, HR and recruiting, Solutions engineering | $201k |
 Anthropic | 1001-5000 | 11 | AI and machine learning | $500k |
 Jump Trading | 1001-5000 | 8 | AI and machine learning, Data science | $300k |
 Lila Sciences | 201-500 | 8 | AI and machine learning | $240k |
 SandboxAQ | 201-500 | 7 | AI and machine learning | $143k |
 Jumio | 201-500 | 6 | AI and machine learning | — |
 Gina's Tech Jobs | 6 | AI and machine learning | $178k | |
 Together AI | 501-1000 | 6 | AI and machine learning | $128k |
 VMAX | 1-10 | 6 | AI and machine learning, Science and research | $400k |
 Sciforium | 201-500 | 6 | AI and machine learning | $203k |
 Astera Institute | 5 | AI and machine learning, Data science | $225k | |
 Riot Games | 1001-5000 | 5 | AI and machine learning, Early careers | — |
 Intrinsic | 51-200 | 5 | Robotics, Backend engineering | — |
 Caylent | 5 | Engineering leadership | — | |
 The Allen Institute for AI | 4 | AI and machine learning, Operations | $199k | |
 Zoox | 1001-5000 | 4 | AI and machine learning | $275k |
 Amgen | 5000+ | 4 | AI and machine learning, Engineering leadership, Management | — |
 CoreWeave | 501-1000 | 4 | AI and machine learning, Product management, Solutions engineering | $212k |
 Citi | 5000+ | 4 | Backend engineering, AI and machine learning | — |
 Boston Dynamics | 501-1000 | 4 | AI and machine learning, Early careers | $178k |
 Tower Research Capital | 501-1000 | 4 | AI and machine learning, DevOps and infrastructure | $250k |
JAX in other countries
What these companies also use
Teams hiring for JAX
Get all 236 companies as a list
Name, website, size, industry and headquarters of every company that requires JAX, through the API (company.getList with technology=JAX and country=US). 236 companies × 0.02 credits = 4.72 credits ($0.47).
Questions
How do you know a company uses JAX?
A company counts when at least one of its open job postings lists JAX as a requirement, not as a nice-to-have. The list is rebuilt every night from the postings on the employers' own careers boards. Staffing agencies, job republishers and the maker of JAX itself are left out.
Which teams hire for JAX?
Of the 554 open roles that require it, most are in AI and machine learning (377), backend engineering (42) and data science (26).
What do JAX roles pay?
306 of the 554 roles state the pay. The middle of their posted ranges is $212k a year, converted to US dollars.
Can I download this list?
Yes, through the API: company.getList with technology=JAX and country=US returns each company's name, website, size, industry and headquarters for 0.02 credits a row (a credit is $0.10). New accounts start with 30 free credits.