JAX Jobs

55 JAX jobs currently listed

Browse 55 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.

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.

Proxima Fusion Posted 1 week ago

Computational Engineer

Munich, Germany - English

Python Git JAX CadQuery Gmsh
Waymo Posted 1 month ago

Staff Machine Learning Engineer

London, United Kingdom - English

Machine Learning Python JAX TensorFlow
£150.000–162.000 /year
Waymo Posted 1 month ago

Machine Learning Engineer

London, United Kingdom - English

Python JAX TensorFlow Reinforcement Learning Deep Learning +2
£120.000–130.000 /year
Perplexity Posted 1 month ago

Member of Technical Staff (AI Inference Engineer)

London, United Kingdom - English

Rust Python CUDA CuTe DSL Triton +4
Nebius Posted 1 month ago

Senior ML Engineer (AI Research)

Amsterdam, Netherlands +3 (Remote) - English

Remote
Python JAX
Nebius Posted 1 month ago Reposted 1 time(s)

HPC System Engineer

Amsterdam, Netherlands - English

Unix/Linux Python Bash CUDA NCCL +2
Scaleai Posted 1 month ago Reposted 7 time(s)

Machine Learning Engineer, Global Public Sector

Doha, Qatar +1 - English

Python PyTorch JAX LLM Machine Learning
Imc Posted 1 month ago

Machine Learning Engineer - Altas Technologies at IMC

Amsterdam, Netherlands - English

Machine Learning Rust C++ Python JAX +2
Stability AI Posted 1 month ago

Research Scientist – Controlled 3D Generation

English

Remote
PyTorch JAX CUDA Score-based methods
AppLike Posted 1 month ago

Senior Machine Learning Engineer (f/m/d)

Hamburg, Germany - English

Hybrid
Machine Learning Python TensorFlow PyTorch JAX
Janestreet Posted 1 month ago

Machine Learning Engineer

London, United Kingdom - English

On-site
Machine Learning PyTorch
Beyondmath Posted 1 month ago

AI Researcher

London, United Kingdom - English

PyTorch JAX TensorFlow Python
Helsing Posted 1 month ago

AI Research Engineer - ML Engineering

Berlin, Germany +2 - English

On-site
Python PyTorch JAX TensorFlow Machine Learning +1
Beyondmath Posted 1 month ago

Machine Learning Engineer

London, United Kingdom - English

Machine Learning Python TensorFlow PyTorch JAX
Anthropic Posted 1 month ago

Research Engineer, Science of Scaling

London, United Kingdom - English

Hybrid
Python Machine Learning LLM Deep Learning
£260.000–630.000 /year
Anthropic Posted 1 month ago

Research Engineer, Pretraining Scaling - London

London, United Kingdom - English

On-site
JAX TPU PyTorch LLM
£260.000–630.000 /year
Anthropic Posted 1 month ago

Research Engineer, Machine Learning (Reinforcement Learning)

London, United Kingdom - English

Hybrid
Python PyTorch TensorFlow JAX Reinforcement Learning +2
£260.000–630.000 /year
Imc Posted 1 month ago

Principal Machine Learning Engineer

Amsterdam, Netherlands +5 - English

On-site
Python CUDA C++ PyTorch TensorFlow +6
$200.000–250.000 /year
DeepL Posted 1 month ago

Staff Research Scientist | Language AI

Cologne, Germany - English

Hybrid
Python PyTorch JAX Machine Learning Machine Translation +1
Miro Posted 1 month ago

Machine Learning Research Engineer

Copenhagen, Denmark +2 (Remote) - English

Remote
Python PyTorch JAX

Frequently Asked Questions

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