JAX Jobs in London

20 JAX positions in London

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

Wayve 3 weeks ago
Hybrid

Research Scientist, Wayve Labs

London, United Kingdom - English

Python PyTorch Transformers MoE Diffusion +5
Graphcore 3 weeks ago Reposted 4 time(s)
On-site

Senior Engineer (ML Engineer)

London, United Kingdom - English

Python PyTorch TensorFlow JAX Linux
Graphcore 3 weeks ago Reposted 5 time(s)
Hybrid

Staff Software Engineer (ML QA)

London, United Kingdom - English

Python C C++ Linux
Graphcore 4 weeks ago Reposted 2 time(s)
On-site

Research Scientist (Visual Generative AI & World Models)

London, United Kingdom +2 - English

Python
Spotify 1 month ago
Hybrid

Senior Staff Machine Learning Engineer, Content Platform

London, United Kingdom - English

Machine Learning
Anthropic 1 month ago
Hybrid

Research Engineer, Machine Learning (RL Velocity)

London, United Kingdom - English

Machine Learning Reinforcement Learning
£370.000–630.000 /year
Waymo 2 months ago

Staff Machine Learning Engineer

London, United Kingdom - English

Machine Learning Python JAX TensorFlow
£150.000–162.000 /year
Waymo 2 months ago

Machine Learning Engineer

London, United Kingdom - English

Python JAX TensorFlow Reinforcement Learning Deep Learning +2
£120.000–130.000 /year
Perplexity 2 months ago

Member of Technical Staff (AI Inference Engineer)

London, United Kingdom - English

Rust Python CUDA CuTe DSL Triton +4
Janestreet 3 months ago Open for 92+ days
On-site

Machine Learning Engineer

London, United Kingdom - English

Machine Learning PyTorch
Beyondmath 3 months ago Open for 92+ days

AI Researcher

London, United Kingdom - English

PyTorch JAX TensorFlow Python
Beyondmath 3 months ago Open for 97+ days

Machine Learning Engineer

London, United Kingdom - English

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

Research Engineer, Pretraining Scaling - London

London, United Kingdom - English

JAX PyTorch TPU LLM
£260.000–630.000 /year
Anthropic 3 months ago Open for 98+ days
Hybrid

Research Engineer, Machine Learning (Reinforcement Learning)

London, United Kingdom - English

Python PyTorch TensorFlow JAX Machine Learning +1
£260.000–630.000 /year
Anthropic 3 months ago Open for 98+ days
Hybrid

Research Engineer, Science of Scaling

London, United Kingdom - English

Python
£260.000–630.000 /year
Olix 4 months ago Open for 138+ days
On-site

Compiler Engineer

London, United Kingdom - English

C C++ TensorFlow PyTorch JAX
£174.000–234.000 /year
Olix 5 months ago Open for 152+ days
On-site

Engineering Manager, Front End Compiler

London, United Kingdom - English

MLIR Shardy TOSA Linalg
£306.000+ /year
Cohere 6 months ago Open for 199+ days
Remote

Senior ML Systems Engineer, Frameworks & Tooling

London, United Kingdom (Remote) - English

JAX Slurm Ray Kubernetes CUDA +4
Olix 10 months ago Open for 322+ days
On-site

Performance Modelling Engineer

London, United Kingdom - English

C++ Python gem5 SystemC PyTorch +2
£211.000+ /year
Cohere 16 months ago Open for 483+ days
Remote

Member of Technical Staff, Training Performance Engineer

London, United Kingdom (Remote) - English

Python CUDA Triton

Frequently Asked Questions

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