Filter By


Save Search

MORE FILTERS

LESS FILTERS

  • Job Types

414 Jobs in JAX framework

___
Full Time

28 Feb 2026

Summary

Our Success Criteria

  • Design, implement, and improve state-of-the-art Reinforcement Learning (RL) and Imitation Learning (IL) algorithms and test them in real-world settings
  • Keep...

Key Skills

AI Robotic systems,Computer Vision (CV),Design,Effective communication skills,JAX framework,Machine learning techniques,PyTorch

___
Full Time

25 Jan 2026

Summary

You are in your final years of a PhD programme in Machine Learning, Statistics, Computer Vision or NLP, and have already published some of your work at major conferences in the field. During your...

Key Skills

Deep Learning,JAX framework,Machine learning techniques,NeurIPS,Optimization,PyTorch

Applied Research Scientist, Perception LLM/VLM (PhD, New Grad)

Waymo

Mountain View, CA, USA; San Francisco, CA, USA

6-8 year

___
Full Time

24 Jan 2026

Summary

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on...

Key Skills

Computer Vision (CV),Deep Learning,Design,JAX framework,Large Language Models - LLMs,Machine learning techniques,NeurIPS,PyTorch,TensorFlow

Applied Research Scientist, 3D Object Detection (PhD New Grad)

Waymo

Mountain View, CA, USA; San Francisco, CA, USA

4-6 year

___
Full Time

24 Jan 2026

Summary

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on...

Key Skills

AI Robotic systems,C++,Computer Vision (CV),Deep Learning,JAX framework,Machine learning techniques,NeurIPS,Python Programming,PyTorch

___
Full Time

24 Jan 2026

Summary

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on...

Key Skills

AI Robotic systems,Deep Learning,JAX framework,Machine learning techniques,NeurIPS,Python Programming,PyTorch,TensorFlow

___
Full Time

23 Jan 2026

Summary

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on...

Key Skills

Aartificial intelligence,C Programming,C++,Deep Learning,Design,Effective communication skills,JAX framework,Machine learning techniques,Python Programming,PyTorch,TensorFlow

___
Full Time

23 Jan 2026

Summary

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on...

Key Skills

C++,Computer Vision (CV),JAX framework,Machine learning techniques,NeurIPS,Python Programming,PyTorch,TensorFlow

2026 Summer Intern, PhD, Perception

Waymo

Mountain View, CA, USA

0-2 year

___
Full Time

23 Jan 2026

Summary

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on...

Key Skills

Design,JAX framework,Machine learning techniques,NeurIPS,Python Programming,PyTorch,TensorFlow

___
Full Time

23 Jan 2026

Summary

Waymo interns partner with leaders in the industry on projects that create impact to the company. We believe learning is a two-way street: applying your knowledge while providing you with...

Key Skills

C++,JAX framework,Machine learning techniques,Python Programming,PyTorch,TensorFlow

___
Full Time

23 Jan 2026

Summary

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on...

Key Skills

AI Robotic systems,JAX framework,Machine learning techniques,PyTorch,TensorFlow