Filter By


Save Search

MORE FILTERS

LESS FILTERS

  • Job Types

417 Jobs in JAX framework

___
Part Time

5 Aug 2026

Summary

About the role

We're hiring a small cohort of graduate research interns to help build the foundations of proactive intelligence.

This is not a traditional internship. You'll own a...

Key Skills

Aartificial intelligence,Deep Learning,JAX framework,Large Language Models - LLMs,Leadership Skill,Machine learning techniques,Optimization,Python Programming,PyTorch

___
Part Time

4 Aug 2026

Summary

About the role

We're hiring a small cohort of graduate research interns to help build the foundations of proactive intelligence.

This is not a traditional internship. You'll own a...

Key Skills

Aartificial intelligence,Deep Learning,JAX framework,Machine learning techniques,Optimization,Python Programming,PyTorch

___
Contract

11 Jul 2026

Summary

Minimum qualifications:

  • PhD degree in a STEM field (e.g., Computer Science, Mathematics, or Statistics) or equivalent practical experience.
  • Experience with one or more general...

Key Skills

Aartificial intelligence,C++,Design,Java Programming,JavaScript,JAX framework,Machine learning techniques,NeurIPS,Python Programming,PyTorch,TensorFlow

___
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