JAX Jobs
JAX is a Python library for high-performance numerical computing and machine learning, providing automatic differentiation, compilation and hardware acceleration. Engineers use it to build and optimise models for GPUs, TPUs and other accelerators.
Browse 83 open JAX jobs from 5 companies including Nvidia, Anthropic, and SpaceXAI. Popular hiring locations include Santa Clara, San Francisco, and Palo Alto.
More filters
JAX Salary Ranges
Pay is quoted per market: the same role is worth several times more in one city than in another, so a single figure spanning countries would describe none of them. 77 of these 83 listings publish a salary. Markets with fewer than 8 disclosed salaries are not shown.
Median yearly pay by country, each in the currency named in its own row
| Country | Median | Middle 50% | Currency | Listings |
|---|---|---|---|---|
| United States | $270,000 | $245,000 – $310,000 | USD | 71 |
Median yearly pay by metro, each in the currency named in its own row
| Metro | Median | Middle 50% | Currency | Listings |
|---|---|---|---|---|
| Santa Clara, United States | $270,000 | $236,000 – $273,000 | USD | 50 |
| San Francisco, United States | $501,000 | $368,000 – $600,000 | USD | 10 |
Seniority and Work Setup for JAX Roles
- Intern
- 7% — 6 of 83 JAX jobs
- Junior
- 2% — 2 of 83 JAX jobs
- Middle
- 20% — 17 of 83 JAX jobs
- Senior
- 75% — 62 of 83 JAX jobs
A role can be open to more than one level, so the shares add up to more than 100%.
- Hybrid
- 16% — 13 of 83 JAX jobs
- On-site
- 84% — 70 of 83 JAX jobs
9 offer visa sponsorship and 3 offer relocation assistance.
Skills Commonly Paired with JAX
- PyTorch
- 99% — 82 of 83 JAX jobs
- AI
- 92% — 76 of 83 JAX jobs
- Python
- 78% — 65 of 83 JAX jobs
- GPU
- 76% — 63 of 83 JAX jobs
- CUDA
- 70% — 58 of 83 JAX jobs
- Deep Learning
- 65% — 54 of 83 JAX jobs
- Machine Learning
- 63% — 52 of 83 JAX jobs
- Algorithms
- 47% — 39 of 83 JAX jobs
Top Companies Hiring for JAX
Employers with the most open JAX jobs
Where JAX Jobs Are Located
Where these JAX jobs are based
83 Open JAX Jobs
Generation time: 0.213