JAX Jobs · 76 Developer & Engineering Roles
New this week: 11 roles
About these roles
Bjak has the most JAX openings at 49, followed by Waymo with 7 and Graphcore with 5. JAX roles pair with PyTorch (108 listings), Python (101), TensorFlow (36), C++, vLLM, CUDA, LLaMA, and Qwen. The page lists 111 jobs, 42 posted in the last week.
2027 Summer Intern, PhD, Machine Learning - Generative World Models & Controllable Simulation
waymo · Internship · Python, PyTorch, JAX
2027 Summer Intern, Machine Learning, MS/PhD, Simulation Realism (World Model)
waymo · Internship · Python, PyTorch, JAX
Staff Forward Deployed Engineer
Suki · Full-time · Python, Vertex AI, PyTorch
Companies hiring
- Bjak37
- waymo11
- Airwallex2
- Institute of Foundation Models2
- Jane Street2
- WRITER2
- Botz1
- Brain Co.1
In-demand skills
- PyTorch74
- Python66
- Embeddings37
- Vector Search37
- TensorFlow18
- C++10
- Ranking Models8
- CUDA7
United Kingdom
Luxembourg
India
United States
Poland
Sweden
Singapore
Spain
Portugal
Germany