26 to 50 of 53 Google JAX Jobs in the UK

with business needs Requirements: • Proven experience taking ML models from concept to production • Strong mathematical background in optimisation, linear algebra, and regularisation techniques • Expertise in ML frameworks (e.g., PyTorch, JAX, TensorFlow) • Experience maintaining organised, reproducible ML research codebases • Passion for staying current with the ML research landscape • A curious and critical mindset, comfortable asking hard questions and challenging assumptions More ❯

with business needs Requirements: • Proven experience taking ML models from concept to production • Strong mathematical background in optimisation, linear algebra, and regularisation techniques • Expertise in ML frameworks (e.g., PyTorch, JAX, TensorFlow) • Experience maintaining organised, reproducible ML research codebases • Passion for staying current with the ML research landscape • A curious and critical mindset, comfortable asking hard questions and challenging assumptions More ❯

with business needs Requirements: • Proven experience taking ML models from concept to production • Strong mathematical background in optimisation, linear algebra, and regularisation techniques • Expertise in ML frameworks (e.g., PyTorch, JAX, TensorFlow) • Experience maintaining organised, reproducible ML research codebases • Passion for staying current with the ML research landscape • A curious and critical mindset, comfortable asking hard questions and challenging assumptions More ❯

with business needs Requirements: • Proven experience taking ML models from concept to production • Strong mathematical background in optimisation, linear algebra, and regularisation techniques • Expertise in ML frameworks (e.g., PyTorch, JAX, TensorFlow) • Experience maintaining organised, reproducible ML research codebases • Passion for staying current with the ML research landscape • A curious and critical mindset, comfortable asking hard questions and challenging assumptions More ❯

with business needs Requirements: • Proven experience taking ML models from concept to production • Strong mathematical background in optimisation, linear algebra, and regularisation techniques • Expertise in ML frameworks (e.g., PyTorch, JAX, TensorFlow) • Experience maintaining organised, reproducible ML research codebases • Passion for staying current with the ML research landscape • A curious and critical mindset, comfortable asking hard questions and challenging assumptions More ❯

Senior Software Engineer, AI Fitness, Health Coach, Fitbit link Copy link corporate_fare Google place London, UK Mid Experience driving progress, solving problems, and mentoring more junior team members; deeper expertise and applied knowledge within relevant area. Apply link Copy link Bachelor's degree or equivalent practical experience. 5 years of experience with data structures or algorithms and software … years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture. Experience using machine learning frameworks (e.g., Tensorflow, PyTorch, or Jax). Experience with machine learning model pretraining, fine-tuning, evaluation, and methodologies. Preferred qualifications: Master's degree or PhD in Computer Science or related technical field. Experience publishing in high … quality machine learning journals. Experience with Python and C++. About the job Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh More ❯

Job Summary As a researcher at Graphcore, you will contribute to the advancement of AI research, investigating new ideas that push the limits on important AI/ML problems. Specialised hardware has been the key driver of the progress of More ❯

We’re looking for a Research Scientist (Machine Learning) to join an ambitious and interdisciplinary team applying AI to transform drug discovery and accelerate the development of life-changing medicines. This is a chance to work on cutting-edge ML More ❯

We’re looking for a Research Scientist (Machine Learning) to join an ambitious and interdisciplinary team applying AI to transform drug discovery and accelerate the development of life-changing medicines. This is a chance to work on cutting-edge ML More ❯

We’re looking for a Research Scientist (Machine Learning) to join an ambitious and interdisciplinary team applying AI to transform drug discovery and accelerate the development of life-changing medicines. This is a chance to work on cutting-edge ML More ❯

We’re looking for a Research Scientist (Machine Learning) to join an ambitious and interdisciplinary team applying AI to transform drug discovery and accelerate the development of life-changing medicines. This is a chance to work on cutting-edge ML More ❯

We’re looking for a Research Scientist (Machine Learning) to join an ambitious and interdisciplinary team applying AI to transform drug discovery and accelerate the development of life-changing medicines. This is a chance to work on cutting-edge ML 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 ❯

At UnlikelyAI, we're looking for a Senior Applied Scientist to join our Applied Science team. This is a high-impact individual contributor position. You'll help drive the end-to-end lifecycle of projects: from identifying opportunities in literature More ❯

Overview At UnlikelyAI, we are building the future of AI: one that is reliable, accurate and transparent. Our neurosymbolic technology harnesses the power of LLMs and generative AI, and combines it with classical symbolic technology to produce hallucination-resistant artificial 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 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 ❯

and build a brand-new intuitive OS and interfaces for orchestrating factories, monitoring machines, controlling production, etc. all with AI. For context, we are a small dynamic team from Google X, Microsoft, Cambridge, MIT, Imperial, BBC R&D to name a few. We work a lot and are all close friends – come join the family! ‍ AI Scientist @ Matta We … reshaping the manufacturing world. ‍ About you A machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down and dirty with More ❯

and build a brand-new intuitive OS and interfaces for orchestrating factories, monitoring machines, controlling production, etc. all with AI. For context, we are a small dynamic team from Google X, Microsoft, Cambridge, MIT, Imperial, BBC R&D to name a few. We work a lot and are all close friends – come join the family! ‍ AI Scientist @ Matta We … reshaping the manufacturing world. ‍ About you A machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down and dirty with More ❯

and build a brand-new intuitive OS and interfaces for orchestrating factories, monitoring machines, controlling production, etc. all with AI. For context, we are a small dynamic team from Google X, Microsoft, Cambridge, MIT, Imperial, BBC R&D to name a few. We work a lot and are all close friends – come join the family! ‍ AI Scientist @ Matta We … reshaping the manufacturing world. ‍ About you A machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down and dirty 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 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 ❯

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 ❯