JAX Jobs

56 JAX jobs currently listed

Browse 56 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

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

Market insights

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

Remote policy

25%
38%
9%

Remote 25%Hybrid 38%On-site 9%

9 of 56 live ads state a salary.

Seniority mix

  • Senior 54%
  • Staff 20%
  • Mid 15%
  • Junior 5%
  • Lead 5%
  • Principal 2%

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.

Waymo 3 days ago
Hybrid

Senior Research Scientist, AI Foundations Recipes

Oxford, United Kingdom +1 - English

Python JAX TensorFlow PyTorch
£123.000–129.000 /year
UKAtomicEnergyAuthority 6 days ago
Hybrid

Lead Data Scientist

Culham, United Kingdom - English

PyTorch JAX Slurm Ray Horovod +3
£71.815+ /year
DeepL 6 days ago
Hybrid

Senior Research Scientist FMTA

London, United Kingdom - English

Python
Onhires 1 week ago Reposted 1 time(s)
Remote

Machine Learning Engineer

Remote (Europe) - English

Python PyTorch JAX
Onhires 1 week ago
Remote

Technical Lead, Machine Learning

Remote (Europe) - English

Python PyTorch JAX Machine Learning LLM
Brainco 1 week ago

Machine Learning Engineer, Applied AI - Deployed

London, United Kingdom - English

Machine Learning PyTorch JAX TensorFlow
Blacksemiconductor 1 week ago

Senior Software Compute Architect (f/m/d)

Aachen, Germany - English

LLVM MLIR Triton PyTorch JAX +2
DeepL 1 week ago

Senior Research Scientist | Multimodal Systems

London, United Kingdom - English

Python PyTorch JAX TensorFlow
DeepL 1 week ago
Hybrid

Senior Research Scientist | Model Steering

London, United Kingdom - English

Python PyTorch JAX TensorFlow
DeepL 1 week ago
Hybrid

Senior Research Scientist | Model Scaling

London, United Kingdom - English

Python PyTorch JAX TensorFlow LoRA +1
Jumio 2 weeks ago
Remote

Machine Learning Engineer

Austria (Remote) - English

PyTorch TensorFlow JAX Airflow Python +8
Graphcore 2 weeks ago
Hybrid

PyTorch Engineer

Bristol, United Kingdom - English

PyTorch Python C++
Graphcore 2 weeks ago Reposted 1 time(s)
Hybrid

AI Research Engineer

London, United Kingdom - English

Python
Graphcore 2 weeks ago Reposted 1 time(s)
Hybrid

Research Scientist (Visual Generative AI & World Models)

London, United Kingdom - English

Python
Jumptrading 3 weeks ago

Campus ML Research Engineer (Full-Time)

London, United Kingdom - English

Python C++ PyTorch JAX TensorFlow +4
Jumptrading 3 weeks ago
On-site

Campus AI Researcher, PhD/Postdoc (Full-Time)

London, United Kingdom - English

Python C++ PyTorch TensorFlow JAX
Graphcore 3 weeks ago Reposted 2 time(s)
Hybrid

Senior Machine Learning Engineer (Large Systems)

Bristol, United Kingdom - English

Python C++
Graphcore 3 weeks ago
Hybrid

Senior Machine Learning Engineer (Large Systems)

London, United Kingdom - English

Python C++
Graphcore 3 weeks ago Reposted 2 time(s)
Hybrid

Senior Machine Learning Engineer (Large Systems)

Cambridge, United Kingdom - English

Machine Learning PyTorch JAX Python C++
Twilio 1 month ago
Remote

Machine Learning Engineer

Ireland (Remote) - English

Python PyTorch TensorFlow JAX AWS +4

Frequently Asked Questions

How many JAX jobs are available right now?
There are currently 56 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.