1 to 25 of 49 Google JAX Jobs in the UK excluding London

system design Required Experience * Strong software engineering background with Python and or C++* Hands-on experience with ML frameworks such as PyTorch, TensorFlow, or JAX* Experience building or maintaining ML infrastructure, training platforms, or inference systems* Solid understanding of distributed systems, parallel computing, and large-scale data processing* Strong ...

e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindset Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible ...

engineers to ship fast in an environment that iterates constantly What they're looking for Strong Python engineering background Experience with frameworks like PyTorch, JAX or TensorFlow Demonstrated ability to scale ML workflows in real-world settings (cloud, GPU clusters, distributed training) Comfortable with MLOps tooling ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...

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 … 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 ...