joblaze

Direct from source · No middlemen

Jax Jobs in United Kingdom

15 open positions · Updated 1 week ago

Average salary (USD/year): 398.3k–845.8k/yr

15 roles Mid 7, Senior 5, Intern 2, Staff 1 3–5 yrs experience Amsterdam, NL +6 more Full-time, Internship newest 1 week ago

Also hiring: PyTorch Python C++ CUDA TensorFlow Triton

15 Jax roles across 5 companies, most in AI/ML

Who is hiring (5 companies)

Graphcore (8) Anthropic (3) Perplexity (2) CuspAI (1) Perplexity AI (1)

Role types

AI/ML (15)

Work arrangement: 6 hybrid · 3 on-site · 6 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 15 of 15 positions

Search with filters →

Applied AI/ML Engineer (Agents) Hybrid

Join CuspAI as an Applied AI/ML Engineer to design intelligent agents for autonomous materials discovery.

CuspAI Amsterdam, NL Published 1 month ago

Active Learning JAX Bayesian optimization 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 5 months ago

JAX Ranking Machine Learning NLP Rust

Flexible on stack

Apply →

Software Engineer - Triton

Join Graphcore's Triton team to enhance AI frameworks and optimize software for cutting-edge machine learning workloads.

Graphcore Gdańsk, Pomeranian Voivodeship, Poland Published 6 months ago

C++ Triton JAX PyTorch TensorFlow

Flexible on stack

Apply →

Software Engineer - Triton

Join Graphcore's Triton team to enhance AI frameworks and improve performance on cutting-edge hardware.

Graphcore Bristol, UK; Gdańsk, Pomeranian Voivodeship, Poland Published 3 weeks ago

C++ Triton JAX PyTorch TensorFlow

Flexible on stack

Apply →

Senior Machine Learning Engineer (Large Systems)

Join Graphcore as a Senior Machine Learning Engineer to advance AI technology on cutting-edge hardware in a dynamic environment.

Graphcore Cambridge, UK Published 9 months ago

NVLink C++ JAX CUDA PyTorch

Flexible on stack

Apply →

Senior Machine Learning Engineer (Large Systems)

Join Graphcore as a Senior Machine Learning Engineer to advance AI technology on cutting-edge hardware in a collaborative environment.

Graphcore Bristol, UK Published 9 months ago

NVLink C++ JAX CUDA PyTorch

Flexible on stack

Apply →

Senior Machine Learning Engineer (Large Systems)

Join Graphcore as a Senior Machine Learning Engineer to advance AI technology on cutting-edge hardware.

Graphcore London, UK Published 9 months ago

NVLink C++ JAX CUDA PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on next-gen AI hardware.

Graphcore Cambridge, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research with a focus on hardware-aware algorithms and impactful implementations.

Graphcore Bristol, UK Published 2 months ago

C++ Triton JAX CUDA PyTorch

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 2 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →

Research Engineer, Machine Learning (RL Velocity) Hybrid Visa

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

JAX AI/ML PyTorch

Apply →

Internship - Search Machine Learning Engineer

Perplexity London Published 5 months ago

JAX Rust PyTorch TensorFlow Python

Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity London Published 5 months ago

NVLink CuTe DSL Triton JAX CUTLASS

Apply →

Research Engineer, Pretraining Scaling - London Visa

Anthropic London, UK £260k–£630k/yr Published 11 months 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 7 months ago

C++ Trio JAX async/concurrent programming Rust

Apply →