JAX Jobs in London

28 JAX positions in London

Browse 28 positions in London that require JAX experience. Each listing includes the full tech stack, salary range, and remote policy.

Janestreet 4 months ago Open for 138+ days
On-site

Machine Learning Engineer

London, United Kingdom - English

Machine Learning PyTorch
Beyondmath 4 months ago Open for 138+ days

AI Researcher

London, United Kingdom - English

PyTorch JAX TensorFlow Python
Beyondmath 4 months ago Open for 143+ days

Machine Learning Engineer

London, United Kingdom - English

Machine Learning Python TensorFlow PyTorch JAX
Anthropic 4 months ago Open for 144+ days
On-site

Research Engineer, Pretraining Scaling - London

London, United Kingdom - English

JAX TPU PyTorch
£260.000–630.000 /year
Anthropic 4 months ago Open for 144+ days
Hybrid

Research Engineer, Machine Learning (Reinforcement Learning)

London, United Kingdom - English

Python PyTorch TensorFlow JAX
£260.000–630.000 /year
Cohere 8 months ago Open for 245+ days
Remote

Senior ML Systems Engineer, Frameworks & Tooling

London, United Kingdom (Remote) - English

JAX Slurm Ray Kubernetes CUDA +4
Hcompany 8 months ago Open for 263+ days Reposted 22 time(s)
Hybrid

Research Engineer, Model Inference & Serving - London

London, United Kingdom - English

Python Rust C++ Go PyTorch +2
Cohere 17 months ago Open for 529+ days
Remote

Member of Technical Staff, Training Performance Engineer

London, United Kingdom (Remote) - English

Python JAX PyTorch CUDA Triton +2

Frequently Asked Questions

How many JAX jobs are there in London?
There are currently 28 open positions in London that list JAX as a required skill. This updates daily.
Do JAX jobs in London require German?
Language requirements vary by role. Each listing shows whether English is sufficient or if German is needed, so you know upfront.
Are JAX jobs in London remote-friendly?
Remote policy varies by company. Each listing shows whether the role is fully remote, hybrid, or on-site. Check individual postings for details.
Can I filter JAX jobs in London by salary?
Yes. Visit the main search page to combine tech stack, location, and minimum salary filters. Salary data is included when available.