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.
Senior Staff Machine Learning Engineer, Content Platform
London, United Kingdom - English
Machine Learning Engineering Manager - Music
London, United Kingdom (Remote) - English
Research Scientist (Embodied AI & World Models)
Bristol, United Kingdom +2 - English
AI Research Scientist
Bristol, United Kingdom - English
Senior ML Engineer
Bristol, United Kingdom - English
2026 Graduate Software Engineer - PyTorch
Bristol, United Kingdom - English
Senior Software Engineer
Cambridge, United Kingdom - English
Senior Machine Learning Engineer (Large Systems)
London, United Kingdom - English
Senior Machine Learning Engineer (Large Systems)
Bristol, United Kingdom - English
2026 Graduate Software Engineer - Triton
Bristol, United Kingdom - English
Software Engineer - Triton
Bristol, United Kingdom +1 - English
Research Scientist (Embodied AI & World Models)
Cambridge, United Kingdom +2 - English
Senior Software Engineer
Bristol, United Kingdom - English
2026 Graduate Machine Learning Engineer - Applied AI
Bristol, United Kingdom - English
Pytorch Engineer
Bristol, United Kingdom - English
Research Engineer, Machine Learning (RL Velocity)
London, United Kingdom - English
Research Engineer - Training Large Behavior Models with Reinforcement Learning (EG16, f/m/div.)
Renningen, Germany (Remote) - English (fluent)
Research Scientist Agentic AI, Reinforcement Learning and Neuro-Symbolic Systems (f/m/div.)
Renningen, Germany - English (fluent)
Research Engineer - Reinforcement Learning and Agentic AI (f/m/div.)
Renningen, Germany - English (fluent)
Master Thesis Bridging the Gap between Reinforcement Learning & E2E Driving
Renningen, Germany - English (very good)
Frequently Asked Questions
- There are currently 55 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.