26 to 34 of 34 Google JAX Jobs in the UK

institutional VCs and prominent angel investors, giving them a solid two-year runway and plans for another upcoming raise. The founding team includes senior engineers and researchers from Meta, Google, and AWS , each with deep experience training and deploying large-scale models. Their mission is simple but ambitious: to develop the next generation of language and multimodal foundation models … data curation, versioning, and reproducible experimentation. You’ll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO More ❯

into their R&D team, building systems that push the limits of computer vision, deep learning, and autonomous vehicle/machinery. You’ll be joining brilliant minds from Tesla, Google DeepMind, NVIDIA and more. 💼 Role: Senior Machine Learning Engineer – Robotics 🌍 Location: London (onsite) 💸 Package: Negotiable + stock options What you’ll be doing Training policies via representation learning, behaviour … looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with More ❯

into their R&D team, building systems that push the limits of computer vision, deep learning, and autonomous vehicle/machinery. You’ll be joining brilliant minds from Tesla, Google DeepMind, NVIDIA and more. 💼 Role: Senior Machine Learning Engineer – Robotics 🌍 Location: London (onsite) 💸 Package: Negotiable + stock options What you’ll be doing Training policies via representation learning, behaviour … looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with More ❯

into their R&D team, building systems that push the limits of computer vision, deep learning, and autonomous vehicle/machinery. You’ll be joining brilliant minds from Tesla, Google DeepMind, NVIDIA and more. 💼 Role: Senior Computer Vision Engineer - Robotics 🌍 Location: London (onsite) 💸 Package: Negotiable + stock options What you’ll be doing Training policies via representation learning, behaviour … looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with More ❯

AI solutions and are hiring exceptional researchers who can bridge the gap between cutting-edge generative AI research and real-world applications. A World-class team with alumni from Google Brain, Meta AI, DeepMind, Stanford, Oxford Unlimited research freedom with strong industry impact What You'll Do Pioneer multimodal AI research combining language models with advanced generative techniques Lead … record Deep expertise in diffusion models and generative AI Proven experience with text-to-image generation, multimodal AI Large-scale model training experience (distributed systems, GPUs) Proficiency in PyTorch, JAX, or similar frameworks Enterprise AI deployment experience Safety and alignment research background Multilingual/multicultural AI expertise Highly Valued: A Background at top-tier AI labs (DeepMind, OpenAI, Meta More ❯

into their R&D team, building systems that push the limits of computer vision, deep learning, and autonomous vehicle/machinery. You’ll be joining brilliant minds from Tesla, Google DeepMind, NVIDIA and more. 💼 Role: Senior Engineer – Computer Vision/Autonomous Machinery 🌍 Location: London (onsite) 💸 Package: Negotiable + stock options What you’ll be doing Training policies via representation … looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with More ❯

into their R&D team, building systems that push the limits of computer vision, deep learning, and autonomous vehicle/machinery. You’ll be joining brilliant minds from Tesla, Google DeepMind, NVIDIA and more. 💼 Role: Senior Engineer – Computer Vision/Autonomous Machinery 🌍 Location: London (onsite) 💸 Package: Negotiable + stock options What you’ll be doing Training policies via representation … looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with More ❯

move from experimentation to full-scale production without the need for a massive ML engineering organisation . An early-stage, high-impact startup founded by engineers and researchers from Google, Spotify, and Uber , backed by some of the world’s most influential investors . With offices in London and San Francisco , they’ve just completed a major funding round … ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. More ❯

move from experimentation to full-scale production without the need for a massive ML engineering organisation . An early-stage, high-impact startup founded by engineers and researchers from Google, Spotify, and Uber , backed by some of the world’s most influential investors . With offices in London and San Francisco , they’ve just completed a major funding round … ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. More ❯