51 to 75 of 93 Google JAX Jobs in the UK

image/video generative models Experience applying reinforcement learning with deep neural networks in practical domains Strong proficiency in Python and PyTorch or JAX, including debugging and profiling numerical code Self-driven with clear communication, documentation, and concise presentation of trade-offs Desirable: experience with robotics simulators (e.g., Isaac ...

models — architecture, training, and inference. Experience solving real problems using reinforcement learning with deep neural networks in any domain. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You are self‐driven, pro‐active, communicate efficiently, document experiments clearly and communicate trade ...

small datasets. Clear writing and communication about modelling choices, results and their limitations, for both ML and scientific audiences. Strong Python and PyTorch or JAX, and daily use of coding agents. NICE TO HAVE Experience post-training LLM agents or protein models with reinforcement learning, for example with ProtRL. ...

develop new tools that are designed to be functional and easy to learn and use. Key Requirements Experience with inference frameworks such as JAX and/or PyTorch Experience of software development in Python, Rust, C/C++, or a similar language Experience monitoring and optimising code written ...

performance, reliability, and operation. Strong Python and generalist software engineering skills across backend services, data systems, and internal tools, with familiarity with PyTorch, JAX, or similar ML frameworks. Experience with distributed training or inference, including diagnosing performance bottlenecks and failures across the stack. Understanding of post-training workflows; experience ...

develop new tools that are designed to be functional and easy to learn and use. Key Requirements Experience with inference frameworks such as JAX and/or PyTorch Experience of software development in Python, Rust, C/C++, or a similar language Experience monitoring and optimising code written ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Knowledge of command-line tools and shell scripting. Knowledge of software engineering best practices, including continuous integration (CI) and continuous deployment … issues in live systems. Experience designing and administering SQL databases — schema design, query performance, and day-to-day operational management. Hands-on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud Storage, Artifact Registry, Cloud SQL. Fluency in English. ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Knowledge of command-line tools and shell scripting. Knowledge of software engineering best practices, including continuous integration (CI) and continuous deployment … issues in live systems. Experience designing and administering SQL databases - schema design, query performance, and day-to-day operational management. Hands-on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud Storage, Artifact Registry, Cloud SQL. Fluency in English. ...

researchers and developers to leverage Graphcore hardware efficiently. You will work on integrating state-of-the-art ML frameworks like PyTorch, Triton, and JAX, while collaborating across teams to deliver performant, reliable software. You’ll contribute to a growing software stack, review code, write documentation, and engage with open ...

team pushing real-time speech technology. Ideal candidates have MSc/PhD in ML or related fields and hands-on experience with PyTorch/JAX/TF, plus several years in research or industry roles. #J-18808-Ljbffr ...

significant experience designing and building production-grade machine learning systems at scale. You have deep experience with modern ML frameworks such as PyTorch, TensorFlow, JAX, or similar. You have experience with—or a strong interest in—multimodal machine learning across areas such as text, audio, image, or video. ...

will work across science, engineering, and product teams to deliver high-impact AI solutions. Ideal candidates are fluent in English, proficient with PyTorch or JAX, write clean Python, and can navigate large codebases independently while collaborating with client teams on complex use cases. #J-18808-Ljbffr ...

data quality, and ensure continuous training cycles. Preferred qualifications include 3+ years in deep learning, proficiency in Python and frameworks like PyTorch/JAX, and effective communication. The role offers 23 days off, private healthcare, and a pension scheme. #J-18808-Ljbffr ...

real-world environments, building robust policies and deployment pipelines. The role requires 3+ years in deep learning for robotics, strong Python and PyTorch or JAX skills, and familiarity with robotics simulators like MuJoCo or Isaac Sim. Onsite, full-time position in London. #J-18808-Ljbffr ...

Ltd. in London invites a Senior Compiler Engineer to help build the ML backend for the DX-1 accelerator. You will connect PyTorch, TensorFlow, JAX and MXNet to our low-level kernel drivers on custom hardware, pushing performance across large AI models. You will own projects, design and implement ...

Bristol, UK About Graphcore At Graphcore, we’re building the future of AI compute.We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon ...

Bristol, UK; Cambridge, UK; London, UK About Graphcore Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation ...

Bristol, UK; Cambridge, UK; London, UK About Graphcore Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation ...

Bristol, UK About Graphcore At Graphcore, we’re building the future of AI compute.We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon ...

Job title: ML Systems Engineer - Model Training and Infrastructure (SWE-focused LLMs) Location: London; full in-office working as default Start date: ASAP Compensation: £80,000 - £110,000 Base Salary & £80,000 - £110,000 Share ...

Bristol, UK; Cambridge, UK; London, UK About Graphcore Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation ...

cross-functionally with internal and external science, engineering, and product teams to deliver high-impact AI solutions. Ideal candidates have experience with PyTorch or JAX, strong Python and a track record from projects or academia. #J-18808-Ljbffr ...

About WRITER WRITER is the AI platform built for the world’s most ambitious marketing and revenue teams, bringing the frontier of AI to the people who drive enterprise growth. With WRITER, autonomous agents work ...

Full TimeOn-siteCambridge, UKPosted 16 hours ago60,000-120,000 GBP/Year Machine Learning Engineer job in Cambridge | Salary: 60,000-120,000Hexwired has partnered with an innovative technology company developing advanced AI and ...

ensuring models stay honest against field conditions and remain deployable in real-time systems. Experience with time-series or biosignals ML using PyTorch or JAX is highly valued, with hybrid work across Manchester and Los Angeles. #J-18808-Ljbffr ...