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.

3 jobs · 2 companies · last seen 10 June 2026

There are currently 3 open roles in Berlin that mention JAX, posted by 2 companies. It frequently pairs with Python, PyTorch, CUDA. Seen most in AI/ML (1% of their roles).

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

By job family

% of open roles in each family that mention JAX

  • AI/ML 1% (3/214)

By company

Top companies looking for JAX

  • distil labs distil labs 2
  • NVIDIA NVIDIA 1

Open roles mentioning JAX

distil labs

distil labs

Machine Learning Engineer

Remote
AI/ML
distil labs

distil labs

Machine Learning Science Intern

Internship Remote
AI/ML