10 of 10 Google JAX Jobs

ML Data Engineer - Healthcare Data Curation & Cleaning (1 Year Fixed Term)

Hiring Organisation
Stanford University
Location
Stanford, California, United States
Employment Type
Permanent
Salary
USD Annual
NumPy). Hands-on experience in building and maintaining automated data pipelines for large-scale data processing. Familiarity with machine learning frameworks (e.g., PyTorch, JAX, scikit-learn) as applied to data quality and augmentation tasks. Expertise in working with healthcare data, including familiarity with the OMOP Common Data Model ...

Machine Learning Research Scientist / Engineer, Reasoning

Hiring Organisation
Scale Ai
Location
Seattle, Washington, United States
Employment Type
Permanent
Salary
USD Annual
advancements into scalable, real-world solutions. Ideally, you'd have: Practical experience working with LLMs, with proficiency in frameworks like PyTorch, JAX, or TensorFlow. You should also be skilled at rapidly interpreting research literature and turning new ideas into working prototypes. A track record of published research ...

2026 Senior Machine Learning Engineer, Multimodal AI, Computer Vision and Graphics - PhD Early Car

Hiring Organisation
Roblox
Location
San Mateo, California, United States
Employment Type
Permanent
Salary
USD Annual
areas: computer vision, multimodal learning, 3D Graphics, or large-scale representation learning. Experience developing and training deep learning models using modern frameworks (PyTorch, TensorFlow, JAX). A strong research track record, evidenced by multiple publications and presentations in top-tier, peer-reviewed venues. Proficiency in one or more programming ...

Senior Staff Machine Learning Engineer, Perception

Hiring Organisation
Waymo
Location
Mountain View, California, United States
Employment Type
Permanent
Salary
USD Annual
Waymo is an autonomous driving technology company with the mission to be the most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver-The World's Most Experienced Driver -to improve access to mobility while saving ...

Founding Machine Learning Engineer

Hiring Organisation
Bjak
Location
Dublin, Ireland
Employment Type
Permanent
Salary
EUR 80,000 - 120,000 Annual
Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

Founding Machine Learning Engineer

Hiring Organisation
Bjak
Location
Etzelwang, Bayern, Germany
Employment Type
Permanent
Salary
EUR 80,000 - 110,000 Annual
Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

Senior AI Research Scientist (m/f/d)

Hiring Organisation
AMLZ Recruiting
Location
Slough, Berkshire, UK
Employment Type
Full-time
AMLZ Recruiting is the executive search brand of AMLZ GmbH, headquartered in Wiesbaden, Germany. We specialize in placing exceptional professionals and executives with leading companies worldwide. Backed by a global network and deep industry expertise ...

Senior AI Research Scientist (m/f/d)

Hiring Organisation
AMLZ Recruiting
Location
London, UK
Employment Type
Full-time
AMLZ Recruiting is the executive search brand of AMLZ GmbH, headquartered in Wiesbaden, Germany. We specialize in placing exceptional professionals and executives with leading companies worldwide. Backed by a global network and deep industry expertise ...

Machine Learning Consultant

Hiring Organisation
Lynx Recruitment Ltd
Location
Melbourn, Royston, Hertfordshire, England, United Kingdom
Employment Type
Full-Time
Salary
Salary negotiable
*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning ...

AI Engineer & Researcher - Multimodal

Hiring Organisation
Xai
Location
Palo Alto, California, United States
Employment Type
Permanent
Salary
USD Annual
knowledge with their teammates. xAI does not have recruiters. Every application is reviewed directly by a technical member of the team. Tech Stack Python JAX and XLA Rust Location The role is based in the Bay Area San Francisco and Palo Alto . Candidates are expected to be located … that can accurately understand the universe and create new knowledge. Ideal Experiences Building distributed training systems for AI models in HPC clusters. Working with JAX/XLA for multi-host training. Digging into third-party source code for debugging and customization. Interview Process After submitting your application, the team ...