1 to 25 of 47 Google JAX Jobs in London

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 hybrid work environment Opportunities to More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 hybrid work environment Opportunities to More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 hybrid work environment Opportunities to More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 hybrid work environment Opportunities to More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this More ❯

AI systems for safe, efficient automation. About You 3+ years of experience building and deploying deep learning systems (industry or research). Strong proficiency in Python and PyTorch or JAX . Hands-on experience with LLMs, VLMs , or generative models for image/video. Deep understanding of training infrastructure (streaming datasets, checkpointing, distributed compute). Strong communicator with clear More ❯

VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for More ❯

VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for More ❯

VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for More ❯

VLMs, or image/video generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics or autonomous driving experience. RL for More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … of high-quality research, evidenced by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. Strong analytical and critical thinking More ❯

+ Discretionary Bonus + Equity We are a stealth-mode AI laboratory researching and developing Machine Learning models. The founding team consists of Cambridge graduates and former engineers at Google, Meta, Bloomberg, Microsoft and Amazon. We are backed by prominent investors from the US and the UK, including institutional VC funds and C-level executives of global technology companies. … models in a distributed environment. LLMs: Understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Proficiency in Python using PyTorch/TensorFlow/JAX framework. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP or Azure), an understanding of containerisation (e.g., Docker). Algorithms and data structures: Excellent More ❯

+ Discretionary Bonus + Equity We are a stealth-mode AI laboratory researching and developing Machine Learning models. The founding team consists of Cambridge graduates and former engineers at Google, Meta, Bloomberg, Microsoft and Amazon. We are backed by prominent investors from the US and the UK, including institutional VC funds and C-level executives of global technology companies. … models in a distributed environment. LLMs: Understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Proficiency in Python using PyTorch/TensorFlow/JAX framework. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP or Azure), an understanding of containerisation (e.g., Docker). Algorithms and data structures: Excellent More ❯

+ Discretionary Bonus + Equity We are a stealth-mode AI laboratory researching and developing Machine Learning models. The founding team consists of Cambridge graduates and former engineers at Google, Meta, Bloomberg, Microsoft and Amazon. We are backed by prominent investors from the US and the UK, including institutional VC funds and C-level executives of global technology companies. … models in a distributed environment. LLMs: Understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Proficiency in Python using PyTorch/TensorFlow/JAX framework. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP or Azure), an understanding of containerisation (e.g., Docker). Algorithms and data structures: Excellent More ❯