JAX Jobs in London

21 JAX positions in London

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

DeepL 1 month ago

Senior Staff Research Scientist | Voice

London, United Kingdom - English

Python PyTorch JAX
DeepL 1 month ago
Hybrid

Senior Research Scientist FMTA

London, United Kingdom - English

Python
DeepL 1 month ago

Senior Research Scientist | Multimodal Systems

London, United Kingdom - English

Python PyTorch JAX TensorFlow
DeepL 1 month ago
Hybrid

Senior Research Scientist | Model Steering

London, United Kingdom - English

Python PyTorch JAX TensorFlow
DeepL 1 month ago
Hybrid

Senior Research Scientist | Model Scaling

London, United Kingdom - English

Python PyTorch JAX TensorFlow LoRA +1
Jumptrading 2 months ago

Campus ML Research Engineer (Full-Time)

London, United Kingdom - English

Python C++ PyTorch JAX TensorFlow +4
Jumptrading 2 months ago
On-site

Campus AI Researcher, PhD/Postdoc (Full-Time)

London, United Kingdom - English

Python C++ PyTorch TensorFlow JAX
Graphcore 2 months ago
Hybrid

Senior Machine Learning Engineer (Large Systems)

London, United Kingdom - English

Python C++
Spotify 4 months ago Open for 142+ days
Hybrid

Senior Staff Machine Learning Engineer - Content Platform

London, United Kingdom - English

Machine Learning
Anthropic 4 months ago Open for 147+ days
Hybrid

Research Engineer, Machine Learning (RL Velocity)

London, United Kingdom - English

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

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

London, United Kingdom - English

Python Rust PyTorch JAX TensorFlow
Perplexity 5 months ago Open for 170+ days

Member of Technical Staff (AI Inference Engineer)

London, United Kingdom - English

Rust Python CUDA CuTe DSL Triton +4
Janestreet 6 months ago Open for 183+ days

Machine Learning Research Engineer

London, United Kingdom - English (fluent)

PyTorch
Beyondmath 6 months ago Open for 183+ days

AI Researcher

London, United Kingdom - English

PyTorch JAX TensorFlow
Beyondmath 6 months ago Open for 188+ days

Machine Learning Engineer

London, United Kingdom - English

Python TensorFlow PyTorch JAX
Anthropic 6 months ago Open for 189+ days
On-site

Research Engineer, Pretraining Scaling - London

London, United Kingdom - English

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

Research Engineer, Machine Learning (Reinforcement Learning)

London, United Kingdom - English

Python PyTorch TensorFlow JAX
£260.000–630.000 /year
Cohere 9 months ago Open for 290+ days Reposted 2 time(s)
Remote

Senior ML Systems Engineer, Frameworks & Tooling

London, United Kingdom (Remote) - English

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

Research Engineer, Model Inference & Serving - London

London, United Kingdom - English

Python Rust C++ Go PyTorch +2
Cohere 11 months ago Open for 352+ days Reposted 5 time(s)

Applied Machine Learning Engineer

London, United Kingdom - English

Python TensorFlow TF-Serving JAX XLA/MLIR

Frequently Asked Questions

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