joblaze

Direct from source · No middlemen

JAX Jobs in London

12 open positions · Updated 1 day ago

Average salary (USD/year): 397.2k–843.5k/yr

12 roles Mid 4, Senior 3, Intern 2, Staff 2 3 yrs experience London, London, UK Full-time, Internship newest 1 day ago

Also hiring: PyTorch Python Kubernetes TensorFlow Rust CUDA

12 JAX roles across 7 companies, most in AI/ML

Who is hiring (7 companies)

Anthropic (3) Graphcore (2) Perplexity (2) Perplexity AI (2) Coreweave (1)

Role types

AI/ML (11) DevOps/SRE (1)

Work arrangement: 4 hybrid · 3 on-site · 5 not stated

Counts are open roles Joblaze currently tracks on company career pages for this exact skill/location; salaries are advertised minimums, annual, converted to USD.

Showing 12 of 12 positions

Search with filters →

Research Engineer, Infrastructure

Join Harvey as a Research Engineer to build infrastructure for ML models in legal work at a fast-scaling company.

Harvey London Published 1 day ago

JAX PyTorch Python

Flexible on stack

Apply →

Platform Engineer

Join CuspAI as a Platform Engineer to build and maintain ML infrastructure that empowers researchers in materials science.

CuspAI London, UK Published 1 week ago

Terraform Kapitan JAX Go PyTorch

Flexible on stack

Apply →

Senior Machine Learning Engineer (Large Systems) Hybrid

Join Graphcore as a Senior Machine Learning Engineer to advance AI models on cutting-edge hardware in a hybrid work environment.

Graphcore London, UK Published 3 weeks ago

C++ JAX CUDA Machine Learning PyTorch

Flexible on stack

Apply →

Internship - Search Machine Learning Engineer

Join Perplexity AI as a Search Machine Learning Engineer Intern to enhance search technologies in a collaborative environment.

Perplexity AI London Published 6 months ago

JAX Ranking Machine Learning NLP Rust

Flexible on stack

Apply →

Member of Technical Staff (AI Inference Engineer)

Join Perplexity AI as an AI Inference Engineer to optimize and develop our inference engine for various model architectures.

Perplexity AI London Published 6 months ago

NVLink CuTe DSL JAX GPU programming NVIDIA

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on innovative AI hardware solutions.

Graphcore London, UK Published 3 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

Account Solution Architect

Join CoreWeave as an Account Solution Architect to support AI labs and enterprises with GPU infrastructure and MLOps solutions across Northern EMEA.

Coreweave London, UK Published 5 months ago

MLOps JAX PyTorch TensorFlow GPU

Flexible on stack

Apply →

Research Engineer, Machine Learning (RL Velocity) Hybrid Visa

Anthropic London, UK £370k–£630k/yr Published 6 months ago

JAX AI/ML PyTorch

Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity London Published 6 months ago

NVLink CuTe DSL Triton JAX CUTLASS

Apply →

Internship - Search Machine Learning Engineer

Perplexity London Published 6 months ago

JAX Rust PyTorch TensorFlow Python

Apply →

Research Engineer, Pretraining Scaling - London Visa

Anthropic London, UK £260k–£630k/yr Published 1 year ago

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Hybrid Visa

Anthropic London, UK £260k–£630k/yr Published 8 months ago

C++ Trio JAX async/concurrent programming Rust

Apply →