joblaze

Direct from source · No middlemen

Jax Jobs in London

9 open positions · Updated 1 month ago

Average salary: 287.5k–630k/yr

Showing 9 of 9 positions

Search with filters →

Internship - Search Machine Learning Engineer

Perplexity London Published 1 month ago

JAX Rust PyTorch TensorFlow Python

UK Internship Program Remote

Perplexity London Published 3 months ago

Torch Triton JAX CUDA AI/ML

Member of Technical Staff (AI Inference Engineer)

Perplexity London Published 1 month ago

NVLink CuTe DSL Triton JAX CUTLASS

Research Scientist, Gemini Diffusion Visa

Google DeepMind London, UK Published 2 months ago

Large-scale Distributed Training Text Diffusion JAX Machine Learning Diffusion Models

Research Scientist, Reinforcement Learning

Google DeepMind London, UK Published 2 months ago

JAX AI/ML reinforcement learning PyTorch Flax

Research Engineer, Machine Learning (RL Velocity) Remote Visa

Anthropic London, UK £370k–£630k/yr Published 1 month ago

JAX AI/ML PyTorch

Research Engineer, Pretraining Scaling - London Visa

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

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

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

C++ Trio JAX async/concurrent programming Rust

Research Engineer, Science of Scaling Remote Visa

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

JAX transformer architectures ETL large-scale ML systems reinforcement learning

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