51 to 70 of 70 Google JAX Jobs in London

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

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

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

About Mistral Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute. We partner with enterprises tackling the hardest problems across high-stakes industries like finance, manufacturing, defense, healthcare, and ...

Compiler Engineer to join the team building the ML backend (compiler, run-time, and debugger) for our next-generation hardware that connect PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large … impact technical and business milestones that drive the success of Olix. Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware. Optimise Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximise throughput. Mentor & Encourage Standards: Lead ...

About Mistral Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute. We partner with enterprises tackling the hardest problems across high-stakes industries like finance, manufacturing, defense, healthcare, and ...

Help us use technology to make a big green dent in the universe! Kraken powers some of the most innovative global developments in energy. We create the technology that redefines utilities and unlocks a new ...

approaches, and lessons learned You May Be a Good Fit If You Have hands-on experience training large language models, or deep expertise with JAX, TPU, PyTorch, or large-scale distributed systems Genuinely enjoy both research and engineering work—you'd describe your ideal split as roughly 50/… Care about the societal impacts of AI and responsible scaling Strong Candidates May Also Have Previous experience training LLM’s or working extensively with JAX/TPU, PyTorch, or other ML frameworks at scale Contributed to open-source LLM frameworks (e.g., open_lm, llm-foundry, mesh-transformer-jax ...

Machine Learning Engineers at IMC play a pivotal role in building the systems that drive training and deployment of large-scale ML models across IMC's global operations. They collaborate with quantitative researchers, hardware specialists ...

About Mistral Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute. We partner with enterprises tackling the hardest problems across high-stakes industries like finance, manufacturing, defense, healthcare, and ...

methods and fidelity levels. Develop and implement solvers and numerical methods, potentially in modern differentiable or GPU-native frameworks such as JAX or NVIDIA Warp. Work with multi-fidelity data: understand tradeoff between fidelity and CFD simulators and how the two can be combined. Use commercial and open-source … ideally in turbomachinery, aerospace, defence or the chemical and process industries (CPI). Experience developing numerical solvers or PDE solvers, for example in JAX, Warp or a comparable framework. Hands‐on experience with at least one commercial CFD solver, such as Simcenter STAR-CCM+ or Ansys Fluent. Comfort working ...

Role: VLA Pre-training Engineer/Deep Learning Engineer – World Models (2 openings) Company: Humanoid Location: London, UK (King’s Cross) – on-site, five days a week Job Type: Full-time About Humanoid Humanoid is ...

Role: Reinforcement Learning Engineer - Manipulation Company: Humanoid Location: London, UK (King's Cross) - on-site, five days a week Job Type: Full-time About Humanoid Humanoid is building the world's most capable, commercially scalable ...

Engineering · London# VLM Research Engineer* Full-timeApply for this roleYou'll take open-weight vision-language models and post-train them for specific intelligence tasks on images and video, owning the work from data to ...

CoreWeave is The Essential Cloud for AITM. Built for pioneers by pioneers, CoreWeave delivers a platform of technology, tools, and teams that enables innovators to build and scale AI with confidence. Trusted by leading AI ...

Mistral AI in the United Kingdom is seeking Applied Scientists and Research Engineers focused on formal methods. You will design, train, and deploy state-of-the-art large language models and apply them to diverse ...

London Fully onsite Full time SC-0A7FC31829C467BD1C22 The opportunity Join a cutting-edge team developing humanoid robots that amplify human potential. Work at the intersection of applied deep learning and robotics to teach robots manipulation ...