JAX Jobs · 110 Developer & Engineering Roles
New this week: 41 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.
Staff ML Engineer, Perception Research
waymo · Full-time · PyTorch, TensorFlow, JAX
Senior Machine Learning Engineer, Perception
PlusAI · Full-time · Python, PyTorch, TensorFlow
Deep Learning Research Intern — Multimodal BEV Perception
PlusAI · Internship · Python, PyTorch, TensorFlow
Companies hiring
- Bjak49
- waymo7
- Graphcore5
- Plumerai4
- Humanoid3
- Jumio3
- riotgames3
- Arondite2
In-demand skills
- PyTorch107
- Python100
- TensorFlow35
- C++22
- vLLM17
- CUDA13
- LLaMA12
- Qwen12
United Kingdom
China
Netherlands
South Korea
Ireland
Spain
Switzerland
Singapore