JAX Jobs in London

15 JAX positions in London

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

Olix 6 days ago

Senior Compiler Engineer

London, United Kingdom - English

C C++ TensorFlow PyTorch JAX +1
Graphcore 3 weeks ago
Hybrid

Senior Machine Learning Engineer (Large Systems)

London, United Kingdom - English

Machine Learning Deep Learning Python C++
Jumptrading 3 months ago Open for 111+ days

Campus ML Research Engineer (Full-Time)

London, United Kingdom - English

Python C++ PyTorch JAX TensorFlow +4
Imc 3 months ago Open for 118+ days Reposted 4 time(s)

Graduate Deep Learning Researcher - London

London, United Kingdom - English

Python Deep Learning
Spotify 6 months ago Open for 187+ days
Hybrid

Senior Staff Machine Learning Engineer - Content Platform

London, United Kingdom - English

Machine Learning
Anthropic 6 months ago Open for 192+ days
Hybrid

Research Engineer, Machine Learning (RL Velocity)

London, United Kingdom - English

Machine Learning
£370.000–630.000 /year
Hcompany 6 months ago Open for 201+ days Reposted 16 time(s)
Hybrid

Research Engineer / Scientist, Post-training & Reinforcement Learning - London

London, United Kingdom - English

Python Rust PyTorch JAX TensorFlow
Perplexity 7 months ago Open for 215+ days

Member of Technical Staff (AI Inference Engineer)

London, United Kingdom - English

Rust Python CUDA CuTe DSL Triton +4
Janestreet 7 months ago Open for 228+ days

Machine Learning Research Engineer

London, United Kingdom - English (fluent)

PyTorch
Beyondmath 7 months ago Open for 228+ days

AI Researcher

London, United Kingdom - English

PyTorch JAX TensorFlow
Beyondmath 7 months ago Open for 233+ days

Machine Learning Engineer

London, United Kingdom - English

Python TensorFlow PyTorch JAX
Anthropic 7 months ago Open for 234+ days
On-site

Research Engineer, Pretraining Scaling - London

London, United Kingdom - English

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

Research Engineer, Machine Learning (Reinforcement Learning)

London, United Kingdom - English

Python PyTorch TensorFlow JAX
£260.000–630.000 /year
Cohere 11 months ago Open for 335+ days Reposted 9 time(s)
Hybrid

Senior ML Systems Engineer, Frameworks & Tooling

London, United Kingdom - English

JAX CUDA NCCL Docker Slurm +4
Hcompany 11 months ago Open for 353+ days Reposted 22 time(s)
Hybrid

Research Engineer, Model Inference & Serving - London

London, United Kingdom - English

Python Rust C++ Go PyTorch +2

Frequently Asked Questions

How many JAX jobs are there in London?
There are currently 15 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.