76 to 100 of 102 Google JAX Jobs

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

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

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

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

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

neuromorphic engineers, and keep models honest against field conditions. What you’ll bring Experience with ML for time-series or biosignals (PyTorch/JAX), model evaluation under noisy real-world data, and a bias toward deployable systems. Apply for Machine Learning Engineer, Neural Decoding — Manchester/Los Angeles #J ...

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

field, or equivalent research experience. 3 years of experience designing, executing, and analyzing empirical research in AI. Experience using machine learning frameworks (e.g., PyTorch, JAX) to run experiments. Experience working with large language model architectures and their internal representations. Experience communicating scientific concepts to technical and non-technical audiences. … advancing the path toward safe, aligned Artificial General Intelligence (AGI). Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high-quality ...

related technical field, or equivalent research experience.3 years of experience designing, executing, and analyzing empirical research in AI.Experience using machine learning frameworks (e.g., PyTorch, JAX) to run experiments.Experience working with large language model architectures and their internal representations.Experience communicating scientific concepts to technical and non-technical audiences.Publication record … models and advancing the path toward safe, aligned Artificial General Intelligence (AGI).Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high ...

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

Campus AI Researcher, PhD/Postdoc (Full-Time) Full-time AI research role for PhD/postdoc candidates working on open-ended ML research, model design, tooling, and infrastructure from concept to production. For PhD ...

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

125,000 - 150,000London AreaMachine Learning Engineers - are you tired of building LLM wrappers I'm working with a well-funded, VC-backed Defence AI start-up building intelligent systems designed to operate in complex ...

design and scale infrastructure that powers the future of AI-driven engineering? Apply now. Responsibilities Design and scale training environments using PyTorch Distributed, JAX, or NVIDIA NeMo across multi-node/multi-GPU clusters. Manage a mixed-compute strategy spanning public clouds (AWS/Azure) and sovereign industrial clouds … that validates physical consistency laws. Skills/Must have Orchestration: Expert-level Kubernetes (AKS/EKS) is essential. ML Frameworks: Strong proficiency in PyTorch, JAX, or NVIDIA NeMo. HPC/Data: Solid experience with Python, Go, and distributed data tools (Dask/Spark). Background: Experience in AI research ...

understanding of LLM inference performance (e.g. KV caching, memory-bound decoding, quantisation). Experience with other AI accelerators (e.g. AMD GPUs, TPUs with JAX and Pallas, or dataflow, wafer-scale and compute-in-memory architectures). Experience with compiler stacks for AI hardware (e.g. MLIR, XLA, TVM). … Experience with hardware/software co-design, or with hardware performance modelling and simulation. Contributions to performance-critical open-source projects (e.g. PyTorch, Triton, JAX, vLLM). A collaborative and supportive work environment The opportunity to have a high impact in a growing organisation Competitive salary package and pension ...

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

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

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