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.

1 job · 1 company · last seen 21 April 2026

There are currently 1 open role in Berlin that mention JAX, posted by 1 company. 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 1 open role in Berlin that mention JAX, posted by 1 company. 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: NVIDIA.
What skills pair well with JAX?
Employers who ask for JAX also frequently require: Python (100%), PyTorch (100%), CUDA (100%), TVM (100%), MLIR (100%).
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% (1/158)

By company

Top companies looking for JAX

  • NVIDIA NVIDIA 1

Open roles mentioning JAX