Google JAX Job Vacancies

51 to 67 of 67 Google JAX Jobs

Senior Machine Learning Engineer - Robotics

City of London, London, United Kingdom
Signify Technology
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 ❯
Posted:

Senior Machine Learning Engineer - Robotics

london, south east england, united kingdom
Signify Technology
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 ❯
Posted:

Senior Machine Learning Engineer - Robotics

slough, south east england, united kingdom
Signify Technology
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 ❯
Posted:

Senior Machine Learning Engineer - Robotics

london (city of london), south east england, united kingdom
Signify Technology
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 ❯
Posted:

AI Engineer

London, South East, England, United Kingdom
Oscar Technology
AI Engineer | 6 Month Contract | Hybrid | £400-450 (IIR35) We're looking for an AI Engineer with deep expertise in data privacy to help us build safe, compliant, and intelligent systems. This role sits at the intersection of AI, data More ❯
Employment Type: Contractor
Rate: £400 - £450 per day
Posted:

AI Engineer

London, United Kingdom
Oscar Associates (UK) Limited
AI Engineer | 6 Month Contract | Hybrid | £400-450 (IIR35) We're looking for an AI Engineer with deep expertise in data privacy to help us build safe, compliant, and intelligent systems. This role sits at the intersection of AI, data More ❯
Employment Type: Contract
Rate: £400 - £450 per day
Posted:

Senior Computer Vision Engineer - Robotics

London Area, United Kingdom
Signify Technology
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 ❯
Posted:

Senior Computer Vision Engineer - Robotics

City of London, London, United Kingdom
Signify Technology
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 ❯
Posted:

Senior Computer Vision Engineer - Robotics

london, south east england, united kingdom
Signify Technology
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 ❯
Posted:

Senior Computer Vision Engineer - Robotics

london (city of london), south east england, united kingdom
Signify Technology
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 ❯
Posted:

Senior Computer Vision Engineer - Robotics

slough, south east england, united kingdom
Signify Technology
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 ❯
Posted:

Senior Researcher (Multimodal AI & Generative Models)

United Kingdom
Hybrid / WFH Options
Intelix.AI
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 ❯
Posted:

Member of Technical Staff

London Area, United Kingdom
Cubiq Recruitment
move from experimentation to full-scale production without the need for a massive ML engineering organisation . An early-stage, high-impact startup founded by engineers and researchers from Google, Spotify, and Uber , backed by some of the world’s most influential investors . With offices in London and San Francisco , they’ve just completed a major funding round … ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. More ❯
Posted:

Member of Technical Staff

City of London, London, United Kingdom
Cubiq Recruitment
move from experimentation to full-scale production without the need for a massive ML engineering organisation . An early-stage, high-impact startup founded by engineers and researchers from Google, Spotify, and Uber , backed by some of the world’s most influential investors . With offices in London and San Francisco , they’ve just completed a major funding round … ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. More ❯
Posted:

Member of Technical Staff

london, south east england, united kingdom
Cubiq Recruitment
move from experimentation to full-scale production without the need for a massive ML engineering organisation . An early-stage, high-impact startup founded by engineers and researchers from Google, Spotify, and Uber , backed by some of the world’s most influential investors . With offices in London and San Francisco , they’ve just completed a major funding round … ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. More ❯
Posted:

Member of Technical Staff

london (city of london), south east england, united kingdom
Cubiq Recruitment
move from experimentation to full-scale production without the need for a massive ML engineering organisation . An early-stage, high-impact startup founded by engineers and researchers from Google, Spotify, and Uber , backed by some of the world’s most influential investors . With offices in London and San Francisco , they’ve just completed a major funding round … ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. More ❯
Posted:

Member of Technical Staff

slough, south east england, united kingdom
Cubiq Recruitment
move from experimentation to full-scale production without the need for a massive ML engineering organisation . An early-stage, high-impact startup founded by engineers and researchers from Google, Spotify, and Uber , backed by some of the world’s most influential investors . With offices in London and San Francisco , they’ve just completed a major funding round … ego engineering culture focused on shipping. What We’re Looking For Deep experience in Applied Machine Learning and Agentic AI systems . Proficiency in modern ML stacks (Python, PyTorch, JAX, Ray, etc.) and production deployment. Proven ability to move fast, ship code, and bridge research with real-world impact. Experience in early-stage or startup environments is a plus. More ❯
Posted:
Google JAX
10th Percentile
£66,000
25th Percentile
£75,000
Median
£95,000
75th Percentile
£97,500