26 to 50 of 62 Google JAX Jobs in England

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 ❯

This temporary role requires an ML Quant Engineer with expertise within an Investment Bank. The position is based in London and involves developing and implementing machine learning models to support financial decision-making. Client Details The hiring organisation is a More ❯

Overview Join Tether and Shape the Future of Digital Finance. At Tether, we're pioneering a global financial revolution. Our solutions enable businesses-from exchanges and wallets to payment processors and ATMs-to integrate reserve-backed tokens across blockchains. By 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 ❯

+ 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. … a distributed environment. LLMs: Strong understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP 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. … a distributed environment. LLMs: Deep understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP 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. … a distributed environment. LLMs: Strong understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP 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. … a distributed environment. LLMs: Deep understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP 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. … a distributed environment. LLMs: Strong understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP 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. … a distributed environment. LLMs: Strong understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP 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. … a distributed environment. LLMs: Deep understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP 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. … a distributed environment. LLMs: Deep understanding of LLM architectures, pre and/or post-training, evaluations and inference. Programming languages: Excellent proficiency in Python using PyTorch/TensorFlow/JAX framework. Customer facing: Experience working directly with customers to solve applied problems, including product integration. Cloud-native technologies: Experience in developing and deploying in cloud platforms (e.g., AWS, GCP 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 ❯

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. … tools, and hardware optimisation techniques Experience with distributed training frameworks (Ray, Dask, PyTorch Lightning, etc.) Expertise with Python and/or C/C++ Development with machine learning frameworks (JAX, Tensorflow, PyTorch etc.) Passion for profiling, identifying bottlenecks, and delivering efficient solutions. Highly Desirable Track record of successfully scaling ML models. Experience writing custom CUDA kernels or XLA operations. 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 ❯