JAX Jobs

46 JAX jobs currently listed

Browse 46 open positions that require JAX experience. Every listing includes the full tech stack, language requirements, remote policy, and salary data, so you can filter with confidence instead of guessing from job descriptions.

By the numbers

  • 22 companies hiring
  • Mostly Senior roles
  • Often paired with Python, PyTorch, Machine Learning, TensorFlow
  • Top locations: London, Bristol, Amsterdam
  • 100% don't require German
  • 14 new in the last 7 days

Market insights

About 10% of live JAX ads are listed as fully remote.

Remote policy

10%
19%
4%

Remote 10%Hybrid 19%On-site 4%

6 of 48 live ads state a salary.

Seniority mix

  • Senior 42%
  • Mid 19%
  • Junior 15%
  • Lead 8%
  • Staff 8%
  • Principal 8%

What you'll find here

Each job includes structured data from the original posting. You can see exactly which technologies are required vs. nice-to-have, whether the role is remote-friendly, and what salary range to expect, all without reading through pages of job descriptions.

Beyondmath 7 months ago Open for 233+ days

Machine Learning Engineer

London, United Kingdom - English

Python TensorFlow PyTorch JAX
Anthropic 7 months ago Open for 234+ days
Hybrid

Research Engineer, Machine Learning (Reinforcement Learning)

London, United Kingdom - English

Python PyTorch TensorFlow JAX
£260.000–630.000 /year
Anthropic 7 months ago Open for 234+ days
On-site

Research Engineer, Pretraining Scaling - London

London, United Kingdom - English

JAX TPU PyTorch
£260.000–630.000 /year
Zencore 8 months ago Open for 264+ days
Remote

Principal Architect, AI/ML

Remote (EU) - English

PyTorch JAX TPU vLLM LangGraph +17
Cohere 11 months ago Open for 335+ days Reposted 9 time(s)
Hybrid

Senior ML Systems Engineer, Frameworks & Tooling

London, United Kingdom - English

JAX CUDA NCCL Docker Slurm +4
Hcompany 11 months ago Open for 353+ days Reposted 22 time(s)
Hybrid

Research Engineer, Model Inference & Serving - London

London, United Kingdom - English

Python Rust C++ Go PyTorch +2

Frequently Asked Questions

How many JAX jobs are available right now?
There are currently 46 open positions that list JAX as a required skill. This number updates daily as new jobs are posted.
How do you know a job requires JAX?
We analyze every job description to identify the specific technologies mentioned. We distinguish between technologies that are required, nice-to-have, or explicitly not used, so you get an accurate picture of each role's tech stack.
Can I filter out jobs that use certain technologies?
Yes. Jobdex supports negative filtering, which lets you exclude specific technologies from your search. For example, you can find JAX jobs that don't use a particular framework or language.
Are salary ranges accurate?
When a company includes salary in their job posting, we use that directly. Otherwise, we parse salary information from the job description. Each salary figure gets a confidence score. High-confidence data is shown prominently, while less certain figures are clearly marked.