How many jobs ask for JAX?

369 open roles at 159 companies mention JAX in the job description itself — extracted from the posting, not from the company's marketing. 7% of them are marked remote by the employer.

What do JAX roles pay?

The median posted base pay is $267K, computed over the 52 JAX roles whose posting states a yearly range we could parse with high confidence. Roles that list no pay, or hourly pay, are excluded.

Is this the company's whole tech stack?

No. Each count is a job posting that names the technology. A company's aggregate stack, built from all its open roles, is on its company page.

TECHNOLOGY · AI & MACHINE LEARNING

JAX jobs

Google's numerical computing library combining NumPy-like syntax with GPU/TPU acceleration and automatic differentiation for ML research. Official docs ↗

Open roles mentioning it: 369
Companies hiring for it: 159
Remote share: 7%
Median base pay: $267K

LIVE ROLES

Newest openings that ask for JAX

Senior Software Engineer, Machine Learning Platform
AirwallexSG - Singapore
Oct 9
Member of Technical Staff — Inference-TPU
RadixArkPalo Alto Office
Oct 9
Member of Technical Staff — Inference-Multimodal & Diffusion
RadixArkPalo Alto Office
Oct 9
Software Engineering Intern (AI/ML) - Summer 2027
RelayRaleigh, NC
Oct 9
Expression of Interest: Machine Learning Engineer
MolocoMenlo Park, California, United States; New York, New York, United States; Seattle, Washington, United States
Oct 8

Sr. Principal Machine Learning Engineer - Central Product Insights
Riot GamesLos Angeles, USA
Oct 8
Staff Machine Learning Engineer, Animation - Singapore Efficiency Team
Riot GamesSingapore
Oct 8
Helix AI Engineer, Training Performance
Figure AISan Jose, CA
Oct 8
Staff AI Inference and Acceleration Engineer
Figure AISan Jose, CA
Oct 8
2027 Summer Intern, PhD, Machine Learning, Planning/Prediction
WaymoMountain View, CA, USA
Oct 8
Machine Learning Engineer
TwilioRemote
Oct 8
Machine Learning Engineer
TwilioRemote
Oct 8

Get JAX job alerts

Search all 369 JAX roles — free account →

WHO'S HIRING

Companies with open roles mentioning JAX

Waymo46 open roles

Thinking Machines Lab16 open roles

Cohere13 open roles

Anthropic12 open roles

Lila Sciences7 open roles

Jumio6 open roles

Together AI6 open roles

Astera Software6 open roles

Intrinsic5 open roles

Isomorphic Labs5 open roles

Airwallex5 open roles

Boston Dynamics5 open roles

HIRED BY DEPARTMENT

Where JAX roles sit

Data & ML188 Engineering151 Operations3 Product2 Customer Success1 Sales1

APPEARS ALONGSIDE

Technologies mentioned in the same job posts

PyTorch×352 Python×269 TensorFlow×187 C++×80 CUDA×69 Docker/Kubernetes×52 Amazon Web Services×44 Transformers×43 Triton×43 TensorRT×39

COMMON ROLES

Job titles that ask for JAX

Machine Learning Engineer8
Senior Machine Learning Engineer5
Ai/Ml Engineering Manager5
Research Scientist4
Staff Machine Learning Engineer4
Research Engineer4
Ai Research Scientist3
Genesis-World: Core Simulation Engine Engineer3

GO DEEPER

Search by this technology

Every technology on this page is extracted from live job descriptions — not self-reported marketing. Filter jobs by JAX or filter companies by tech stack.

Frequently asked questions

How many jobs ask for JAX?: 369 open roles at 159 companies mention JAX in the job description itself — extracted from the posting, not from the company's marketing. 7% of them are marked remote by the employer.
What do JAX roles pay?: The median posted base pay is $267K, computed over the 52 JAX roles whose posting states a yearly range we could parse with high confidence. Roles that list no pay, or hourly pay, are excluded.
Is this the company's whole tech stack?: No. Each count is a job posting that names the technology. A company's aggregate stack, built from all its open roles, is on its company page.