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

  • 24 companies hiring
  • Mostly Senior roles
  • Often paired with Python, PyTorch, TensorFlow, Machine Learning
  • Top locations: London, Bristol, Amsterdam
  • 100% don't require German

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.

Wayve 3 weeks ago
Hybrid

Research Scientist, Wayve Labs

London, United Kingdom - English

Python PyTorch Transformers MoE Diffusion +5
Graphcore 3 weeks ago Reposted 2 time(s)
On-site

Senior Machine Learning Engineer (Large Systems)

Bristol, United Kingdom - English

Python C++
Graphcore 3 weeks ago Reposted 2 time(s)
On-site

Senior Software Engineer (ML QA)

Cambridge, United Kingdom - English

Python C C++ Linux
Graphcore 3 weeks ago Reposted 4 time(s)
On-site

Senior Engineer (ML Engineer)

London, United Kingdom - English

Python PyTorch TensorFlow JAX Linux
Graphcore 3 weeks ago Reposted 4 time(s)
Hybrid

Senior Engineer (ML Engineer)

Cambridge, United Kingdom - English

Python Linux
Graphcore 3 weeks ago Reposted 3 time(s)
On-site

Staff Engineer (ML Engineer)

Cambridge, United Kingdom - English

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

Staff Engineer (ML Engineer)

Bristol, United Kingdom - English

Machine Learning Python Linux
Graphcore 3 weeks ago Reposted 5 time(s)
Hybrid

Staff Software Engineer (ML QA)

London, United Kingdom - English

Python C C++ Linux
Graphcore 3 weeks ago Reposted 3 time(s)
On-site

Staff Software Engineer (ML QA)

Cambridge, United Kingdom - English

Python C C++ Linux
Graphcore 4 weeks ago Reposted 2 time(s)
On-site

Research Scientist (Visual Generative AI & World Models)

London, United Kingdom +2 - English

Python
Nebius 1 month ago
Remote

ML Infrastructure Engineer

Amsterdam, Netherlands +2 (Remote) - English

PyTorch JAX Megatron-LM TensorRT-LLM CUDA +3
Graphcore 1 month ago
On-site

Senior Research Engineer

Bristol, United Kingdom +2 - English

Python C++ CUDA Triton XLA +1
Spotify 1 month ago
Hybrid

Senior Staff Machine Learning Engineer, Content Platform

London, United Kingdom - English

Machine Learning
Graphcore 1 month ago

Software Engineer - Triton

Bristol, United Kingdom +1 - English

Python C++ Triton PyTorch
Graphcore 1 month ago
Hybrid

2026 Graduate Software Engineer - PyTorch

Bristol, United Kingdom - English

PyTorch Python
Graphcore 1 month ago
Hybrid

2026 Graduate Software Engineer - Triton

Bristol, United Kingdom - English

Python C++ Triton
Graphcore 1 month ago
Hybrid

Pytorch Engineer

Bristol, United Kingdom - English

PyTorch Python C++
Anthropic 1 month ago
Hybrid

Research Engineer, Machine Learning (RL Velocity)

London, United Kingdom - English

Machine Learning Reinforcement Learning
£370.000–630.000 /year
Bosch 1 month ago
Remote

Research Engineer - Training Large Behavior Models with Reinforcement Learning (EG16, f/m/div.)

Renningen, Germany (Remote) - English (fluent)

Python PyTorch TensorFlow JAX Isaac Sim +6
Bosch 1 month ago
Hybrid

Research Scientist Agentic AI, Reinforcement Learning and Neuro-Symbolic Systems (f/m/div.)

Renningen, Germany - English (fluent)

Python PyTorch TensorFlow JAX Machine Learning +3

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.