JAX is a Python library for high-performance numerical computing and automatic differentiation, often used to implement and accelerate machine learning research code on accelerators.
10 jobs · 8 companies · last seen 7 October 2026
There are currently 10 open roles in Berlin that mention JAX, posted by 8 companies. It frequently pairs with PyTorch, Machine Learning, LLM. Seen most in AI/ML (4% of their roles) and genAI/LLM/Agentic AI (3% of their roles). Across those roles, 100% list it as required and 0% as nice-to-have.
Frequently asked questions
- How many JAX jobs are available in Berlin right now?
- There are currently 10 open roles in Berlin that mention JAX, posted by 8 companies. The board is refreshed multiple times a day — more frequently during Berlin working hours — so new roles can appear throughout the day.
- Which companies in Berlin are hiring for JAX?
- Companies currently listing the most JAX roles include: Helsing, LawZero, NVIDIA, Phantom, CuspAI.
- What skills pair well with JAX?
- Employers who ask for JAX also frequently require: PyTorch (100%), Machine Learning (90%), LLM (80%), Python (70%), Deep Learning (70%).
- In which Berlin job roles is JAX most in demand?
- Based on current listings, JAX appears most in: AI/ML (4%), genAI/LLM/Agentic AI (3%) roles.
Frequently paired with
By seniority
- Mid 50% (5)
- Senior 30% (3)
- Staff / Principal 10% (1)
- Intern / Student 10% (1)
By job family
% of open roles in each family that mention JAX
- AI/ML 4% (4/95)
- genAI/LLM/Agentic AI 3% (6/231)
By company
Top companies looking for JAX
-
Helsing 2 -
LawZero 2 -
NVIDIA 1 -
Phantom 1 -
CuspAI 1 -
Hugging Face 1 -
Dunia 1 -
distil labs 1
Open roles mentioning JAX
AI/ML
LawZero
Senior ML Research Scientist
genAI/LLM/Agentic AI
LawZero
Senior ML Research Developer
AI/ML
AI/ML
AI/ML
genAI/LLM/Agentic AI
genAI/LLM/Agentic AI
genAI/LLM/Agentic AI
genAI/LLM/Agentic AI
genAI/LLM/Agentic AI
No jobs match your filters.