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.

DeepL 6 days ago
Hybrid

Senior Research Scientist FMTA

London, United Kingdom - English

Python
Brainco 1 week ago

Machine Learning Engineer, Applied AI - Deployed

London, United Kingdom - English

Machine Learning PyTorch JAX TensorFlow
DeepL 1 week ago

Senior Research Scientist | Multimodal Systems

London, United Kingdom - English

Python PyTorch JAX TensorFlow
DeepL 1 week ago
Hybrid

Senior Research Scientist | Model Steering

London, United Kingdom - English

Python PyTorch JAX TensorFlow
DeepL 1 week ago
Hybrid

Senior Research Scientist | Model Scaling

London, United Kingdom - English

Python PyTorch JAX TensorFlow LoRA +1
Graphcore 2 weeks ago Reposted 1 time(s)
Hybrid

AI Research Engineer

London, United Kingdom - English

Python
Graphcore 2 weeks ago Reposted 1 time(s)
Hybrid

Research Scientist (Visual Generative AI & World Models)

London, United Kingdom - English

Python
Jumptrading 3 weeks ago

Campus ML Research Engineer (Full-Time)

London, United Kingdom - English

Python C++ PyTorch JAX TensorFlow +4
Jumptrading 3 weeks ago
On-site

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

London, United Kingdom - English

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

Senior Machine Learning Engineer (Large Systems)

London, United Kingdom - English

Python C++
Graphcore 1 month ago Reposted 3 time(s)
Hybrid

Staff Engineer (ML Engineer)

London, United Kingdom - English

PyTorch TensorFlow JAX Triton Python +1
Waymo 1 month ago

Staff Machine Learning Engineer

London, United Kingdom - English

Machine Learning Python JAX TensorFlow LLM +1
£155.000–163.000 /year
Wayve 2 months ago
Hybrid

Research Scientist, Wayve Labs

London, United Kingdom - English

Computer Vision Machine Learning PyTorch Python Reinforcement Learning
Spotify 3 months ago Open for 96+ days
Hybrid

Senior Staff Machine Learning Engineer, Content Platform

London, United Kingdom - English

Machine Learning
Anthropic 3 months ago Open for 102+ days
Hybrid

Research Engineer, Machine Learning (RL Velocity)

London, United Kingdom - English

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

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

London, United Kingdom - English

Python Rust PyTorch JAX TensorFlow
Olix 4 months ago Open for 123+ days
On-site

Architect/Staff Systems Software Engineer

London, United Kingdom - English

C C++ PyTorch JAX
Waymo 4 months ago Open for 124+ days

Machine Learning Engineer

London, United Kingdom - English

Python JAX TensorFlow Reinforcement Learning Deep Learning +2
£120.000–130.000 /year
Waymo 4 months ago Open for 124+ days

Staff Machine Learning Engineer

London, United Kingdom - English

Machine Learning Python JAX TensorFlow
£150.000–162.000 /year
Perplexity 4 months ago Open for 125+ days

Member of Technical Staff (AI Inference Engineer)

London, United Kingdom - English

Rust Python CUDA CuTe DSL Triton +4

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.