Latest AI JobsBrowse Jobs

Jax Jobs

26 open roles, updated daily

Filters3
26 jobs foundClear all
W
2027 Summer Intern, PhD, Research, AV Planning
Waymo
9d ago

Apply for the 2027 Summer Intern, PhD, Research, AV Planning role at Waymo in Mountain View, CA. Hybrid onsite internship for doctoral students.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$85 — $85 USDView Job
W
2027 Summer Intern, PhD, Research, Post Training
Waymo
9d ago

Waymo seeks a PhD intern in Mountain View or San Francisco for research in post-training autonomous driving AI. $85/hour hourly pay.

Mountain View, CA, USA: San Francisco, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PyTorchJAXReinforcement Learning
$85 — $85 USDView Job
W
2027 Summer Intern, PhD, Research, World Modeling Evaluation
Waymo
10d ago

Apply for the 2026 Summer Intern, PhD, Research, World Modeling Evaluation role at Waymo in Mountain View, CA. Hybrid onsite PhD internship.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$85 — $85 USDView Job
W
2027 Summer Intern, PhD, Machine Learning, Planning/Prediction
Waymo
11d ago

Apply for the 2027 Summer ML Internship at Waymo in Mountain View, CA. Hybrid PhD role focusing on planning and prediction at $85/hour.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$85 — $85 USDView Job
W
2027 Summer Intern, MS/PhD, Research, Perception Foundation Models
Waymo
12d ago

Apply for the 2027 Summer Intern, Perception Foundation Models role at Waymo in Mountain View, CA. Hybrid onsite graduate machine learning internship.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$70 — $70 USDView Job
Lyft logo
PhD Machine Learning Software Engineer Intern (Summer 2027)
Lyft
12d ago

Apply for the PhD Machine Learning Software Engineer Intern role at Lyft in San Francisco, CA. Summer 2027 hybrid internship focusing on applied AI research.

San Francisco, CAEarly TalentInternshipInternOpen · checked 7d ago
PythonPyTorchTensorFlow
$65-$68View Job
W
2027 Summer Intern, PhD, Research, World Action Model
Waymo
16d ago

Research and implement world action models for autonomous driving at Waymo. Ideal for a Ph.D. candidate with a strong ML background.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$85 — $85 USDView Job
W
2027 Summer Intern, MS/PhD, Machine Learning, Driver Refinement Foundations
Waymo
18d ago

Join Waymo as a 2027 Summer Intern in Machine Learning for Driver Refinement Foundations. Work on LLMs, generative models, and autonomous driving tech.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$70 — $70 USDView Job
W
2027 Summer Intern, PhD, Planner Machine Learning
Waymo
20d ago

Join Waymo as a PhD Planner Machine Learning Intern in San Francisco. Research deep learning architectures for autonomous driving. Apply now.

San Francisco, CaliforniaPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$85 — $85 USDView Job
W
2027 Summer Intern, PhD, Machine Learning Research, Planning/Prediction
Waymo
20d ago

Join Waymo as a PhD Machine Learning Research Intern in Mountain View. Research generative models for autonomous driving planning and prediction.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$85 — $85 USDView Job
W
2027 Summer Intern, MS/PhD, Perception, Machine Learning
Waymo
23d ago

Join Waymo in Mountain View as a Perception Machine Learning Intern. Design multi-modality models using large-scale driving data.

Mountain View, CA, USAPerception (7LT)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$70 — $70 USDView Job
W
2027 Summer Intern, MS/PhD, Software Engineer, Sys Intel & Machine Learning
Waymo
23d ago

Join Waymo as a Summer Intern in Software Engineering for Sys Intel & Machine Learning in Mountain View, CA. Help optimize autonomous driving models.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$70 — $70 USDView Job
W
2027 Summer Intern, PhD, Software Engineer, Simulation
Waymo
25d ago

Join Waymo as a PhD Software Engineering Intern in Simulation. Work on uncertainty-aware algorithms in a hybrid role based in Mountain View.

Mountain View, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonTensorFlowJAX
$85 — $85 USDView Job
W
2027 Summer Intern, MS/PhD, Machine Learning Engineer
Waymo
26d ago

Join Waymo as a Machine Learning Engineering Intern in San Francisco. Train multi-task transformers on TPUs for autonomous driving.

San Francisco, CA, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$70 — $70 USDView Job
W
2027 Summer Intern, MS/PhD, Software Engineer, Simulation Evaluation ML Model
Waymo
27d ago

Join Waymo as a 2027 Summer Intern in Simulation Evaluation ML. Build semantic extraction pipelines and temporal graph models for autonomous driving.

Mountain View, California, USAPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
$70 — $70 USDView Job
W
2027 Summer Intern, PhD, Learning-Based Behavior for Special Vehicles
Waymo
31d ago

Apply for the 2027 Summer Intern, PhD, Learning-Based Behavior for Special Vehicles role at Waymo in Mountain View, California.

Mountain View, CaliforniaPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$85 — $85 USDView Job
Together AI logo
Research Intern, Frontier Agents (Summer 2027)
Together AI
31d ago

Join Together AI as a Research Intern for Frontier Agents in San Francisco. Work on agentic AI, alignment, and scaling with a Ph.D.

San FranciscoResearchInternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$58 - $70View Job
W
2027 Summer Intern, MS/PhD, Sim-Realism ML Infrastructure
Waymo
31d ago

Join Waymo in London as a Sim-Realism ML Infrastructure Intern. Work on distributed systems, Python, C++, and autonomous driving technology.

London, England, United KingdomPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonTensorFlowJAX
£65K — £65K GBPView Job
Together AI logo
Research Intern, Frontier Agents (Winter 2027)
Together AI
31d ago

Join Together AI as a Research Intern for Winter 2027 in San Francisco. Work on frontier agentic systems, reinforcement learning, and LLM alignment.

San FranciscoResearchInternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$58 - $70View Job
Together AI logo
Research Intern, Model Shaping (Winter 2027)
Together AI
31d ago

Join Together AI as a Research Intern for Winter 2027 in San Francisco or Amsterdam. Work on model shaping, post-training, and efficient training.

San Francisco, AmsterdamResearchInternshipInternshipOpen · checked 7d ago
PyTorchJAXNLP
$58 to $70.View Job
Together AI logo
Research Intern, Model Shaping (Summer 2027)
Together AI
31d ago

Join Together AI as a Research Intern for Model Shaping in San Francisco or Amsterdam. Work on post-training, efficient training, and foundation models.

San Francisco, AmsterdamResearchInternshipInternshipOpen · checked 7d ago
PyTorchJAXNLP
$58 to $70.View Job
Together AI logo
Research Intern, Inference (Winter 2027)
Together AI
31d ago

Join Together AI as a Winter 2027 Inference Research Intern in San Francisco. Build scalable serving systems for foundation models. Pay: $58-$70/hr.

San FranciscoResearchInternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$58 to $70View Job
Together AI logo
Research Intern, Inference (Summer 2027)
Together AI
31d ago

Join Together AI as a Summer 2027 Inference Research Intern in San Francisco. Build scalable serving systems for large foundation models.

San FranciscoResearchInternshipInternshipOpen · checked 7d ago
PythonPyTorchJAX
$58 to $70View Job
W
2027 Summer Intern, MS/PhD, Machine Learning, Simulation Realism
Waymo
32d ago

Join Waymo in London as a Machine Learning Simulation Realism Intern. Contribute to autonomous driving technology with Python and deep learning frameworks.

London, England, United KingdomPipeline (N/A)InternshipInternshipOpen · checked 7d ago
PythonPyTorchTensorFlow
£65K — £65K GBPView Job