JAX Jobs
Explore JAX roles on Inference Jobs and apply today.
4w agoGR
2026 Graduate Machine Learning Engineer - Applied AI
Graphcore
Bristol, England, United Kingdom (On-site)
2mo agoNV
Senior Compiler Engineer, AI Inference Platforms
NVIDIA
Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
2w agoAN
Performance Engineer, GPU
Anthropic
San Francisco, California, United States (Hybrid)$280K – $850K Yearly
4w agoSC
Machine Learning Research Scientist / Engineer, Reasoning
Scale
San Francisco, California, United States (On-site)$252K – $315K Yearly
3mo agoXA
Member of Technical Staff, Recommendation Systems
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
1mo agoXA
Software Engineer, ML & Data Infra
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
2mo agoNV
Senior Compiler Engineer, AI Inference Performance
NVIDIA
Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
2w agoNV
1mo agoNV
Senior Deep Learning Algorithm Engineer, Training Framework
NVIDIA
Santa Clara, California, United States (Hybrid)$184K – $287.5K Yearly
1mo agoNV
Senior HPC Performance Engineer - AI for Science at Scale
NVIDIA
Santa Clara, California, United States (On-site)$184K – $287.5K Yearly
1mo agoNV
AI Inference Performance Engineer - New College Grad 2026
NVIDIA
Santa Clara, California, United States (On-site)$124K – $241.5K Yearly