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
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%
Top hiring companies
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.
AI and Computer Vision Engineer
Munich, Germany - English (working proficiency)
PhD - Scalable and Efficient Reinforcement Learning Methods for Physical AI
Renningen, Germany - English (fluent)
Software Engineer - Triton
Bristol, United Kingdom +1 - English
Senior Staff Research Scientist | Voice
London, United Kingdom - English
Senior ML Engineer (AI Research, Physical AI)
Amsterdam, Netherlands +2 (Remote) - English (Excellent)
Senior Research Scientist, AI Foundations Recipes
Oxford, United Kingdom +1 - English
Technical Lead, Machine Learning
Remote (Europe) - English
Senior Software Compute Architect (f/m/d)
Aachen, Germany - English
Senior Research Scientist | Multimodal Systems
London, United Kingdom - English
Senior Research Scientist | Model Steering
London, United Kingdom - English
Senior Research Scientist | Model Scaling
London, United Kingdom - English
Machine Learning Engineer
Austria (Remote) - English
Senior ML Research Engineer, Virtual Cell
United States - English
AI Research Engineer
Cambridge, United Kingdom - English (Professional/Fluent)
Campus ML Research Engineer (Full-Time)
London, United Kingdom - English
Campus AI Researcher, PhD/Postdoc (Full-Time)
London, United Kingdom - English
Senior Machine Learning Engineer (Large Systems)
Bristol, United Kingdom - English
Frequently Asked Questions
- There are currently 45 open positions that list JAX as a required skill. This number updates daily as new jobs are posted.
- 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.
- 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.
- 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.