joblaze

Direct from source · No middlemen

Jax Jobs in London

7 open positions · Updated 3 days ago

Average salary: 260k–630k/yr

Showing 7 of 7 positions

Search with filters →
Perplexity
Internship - Search Machine Learning Engineer
Perplexity
JAX Rust PyTorch TensorFlow Python
Indexed 3 days ago Apply →
Google DeepMind
SWE - Grids - Fixed Term Contract - 6 Months - London, UK
Google DeepMind
C++ Google Cloud Platform JAX TPU Python
Published 2 weeks ago Apply →
Perplexity
UK Internship Program Remote
Perplexity
Torch Triton JAX CUDA AI/ML
Indexed 3 days ago Apply →
Perplexity
Member of Technical Staff (AI Inference Engineer)
Perplexity
NVLink CuTe DSL Triton JAX CUTLASS
Indexed 3 days ago Apply →
Google DeepMind
Research Scientist, Gemini Diffusion Visa
Google DeepMind
Large-scale Distributed Training Text Diffusion JAX Machine Learning Diffusion Models
Published 1 month ago Apply →
Google DeepMind
Research Scientist, Reinforcement Learning
Google DeepMind
JAX AI/ML reinforcement learning PyTorch Flax
Published 1 month ago Apply →
Anthropic
Research Engineer, Pretraining Scaling - London Visa
Anthropic
JAX large-scale distributed systems PyTorch LLM evaluation infrastructure
Indexed 3 days ago Apply →

Related searches

jax Remote jax in New York jax in San Francisco jax in Berlin jax in Amsterdam Python in London Ruby On Rails in London React in London Devops in London Machine Learning in London Golang in London Typescript in London