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.
Machine Learning Research Engineer
Copenhagen, Denmark +2 (Remote) - English
Lead Research Scientist
Copenhagen, Denmark +2 (Remote) - English
AI Research Engineer - Foundation Models
Berlin, Germany +3 - English
Member of Engineering (Pre-training / CUDA)
Remote (7 countries) - English
Machine Learning Engineer
English
Principal Architect, AI/ML
Remote (28 countries) - English
Compiler Engineer
London, United Kingdom - English
Senior Machine Learning Engineer (Computer Vision)
Tampere, Finland (Remote) - English
Engineering Manager, Front End Compiler
London, United Kingdom - English
Member of Engineering (Scalability)
Remote (EMEA) - English
(Senior) Staff Research Scientist | Voice
Munich, Germany - English
Senior ML Systems Engineer, Frameworks & Tooling
London, United Kingdom (Remote) - English
AI Scientist - Zurich
Zurich, Switzerland - English
Member of Engineering (Pre-training and inference software)
Remote (9 countries) - English
Frequently Asked Questions
- There are currently 56 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.