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.

9 jobs · 7 companies · last seen 20 August 2026

There are currently 9 open roles in Berlin that mention JAX, posted by 7 companies. It frequently pairs with Python, Machine Learning, Deep Learning. Seen most in AI/ML (3% of their roles).

Frequently asked questions
How many JAX jobs are available in Berlin right now?
There are currently 9 open roles in Berlin that mention JAX, posted by 7 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, NVIDIA, Nebius, CuspAI, Hugging Face.
What skills pair well with JAX?
Employers who ask for JAX also frequently require: Python (89%), Machine Learning (89%), Deep Learning (78%), PyTorch (78%), LLM (67%).
In which Berlin job roles is JAX most in demand?
Based on current listings, JAX appears most in: AI/ML (3%) roles.

By job family

% of open roles in each family that mention JAX

  • AI/ML 3% (9/316)

By company

Top companies looking for JAX

  • Helsing Helsing 2
  • NVIDIA NVIDIA 2
  • Nebius Nebius 1
  • CuspAI CuspAI 1
  • Hugging Face Hugging Face 1
  • Cohere Cohere 1
  • distil labs distil labs 1

Open roles mentioning JAX

Nebius
AI/ML
20 Aug
Helsing
AI/ML
20 Aug
Helsing
AI/ML
20 Aug
CuspAI
AI/ML
13 Aug
Hugging Face
AI/ML
29 May
Cohere
AI/ML
30 Sept
distil labs

distil labs

Machine Learning Science Intern

Internship Remote
AI/ML