JAX Jobs

45 JAX jobs currently listed

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

  • 26 companies hiring
  • Mostly Senior roles
  • Often paired with Python, PyTorch, TensorFlow, C++
  • Top locations: London, Amsterdam, Berlin
  • 100% don't require German
  • 2 new in the last 7 days

Market insights

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

Remote policy

29%
31%
7%

Remote 29%Hybrid 31%On-site 7%

5 of 45 live ads state a salary.

Seniority mix

  • Senior 62%
  • Mid 18%
  • Staff 9%
  • Junior 6%
  • Lead 3%
  • Principal 3%

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.

Zencore 7 months ago Open for 219+ days
Remote

Principal Architect, AI/ML

Remote (EU) - English

PyTorch JAX TPU vLLM LangGraph +17
Cohere 9 months ago Open for 290+ days Reposted 2 time(s)
Remote

Senior ML Systems Engineer, Frameworks & Tooling

London, United Kingdom (Remote) - English

JAX Slurm Ray Kubernetes CUDA +4
Hcompany 10 months ago Open for 308+ days Reposted 22 time(s)
Hybrid

Research Engineer, Model Inference & Serving - London

London, United Kingdom - English

Python Rust C++ Go PyTorch +2
Cohere 11 months ago Open for 352+ days Reposted 5 time(s)

Applied Machine Learning Engineer

London, United Kingdom - English

Python TensorFlow TF-Serving JAX XLA/MLIR
Cohere 19 months ago Open for 574+ days Reposted 15 time(s)
Remote

Member of Technical Staff, Training Performance Engineer

London, United Kingdom (Remote) - English

Python JAX PyTorch XLA MLIR +2

Frequently Asked Questions

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