JAX Jobs · 35 Developer & Engineering 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.
Research Scientist, Optimization Systems
Aalyria · Full-time · Python, PyTorch, TensorFlow
United States3w ago
Senior Research Engineer
AssemblyAI · Full-time · JAX, XLA, Pallas
AI Security Researcher/Research Engineer
Brave · Full-time · Python, PyTorch, JAX
Companies hiring
- Bjak25
- Airwallex2
- Aalyria1
- AfterQuery1
- AssemblyAI1
- Baseten1
- Brave1
- CuspAI1
In-demand skills
- PyTorch34
- Python29
- OpenAI12
- Ray5
- TensorRT-LLM5
- vLLM5
- Apache Arrow3
- DeepSpeed3
United Kingdom
Australia
Singapore
South Korea
China
Germany
Netherlands
Switzerland
Ireland
Sweden