JAX Jobs
Browse 137 JAX jobs on Inference Jobs.
137 jobs
3wXA
Member of Technical Staff, JAX & Compiler
xAI
Palo Alto, California, United States (On-site)$180k – $440k Yearly
3wXA
Member of Technical Staff, Midtraining
xAI
Palo Alto, California, United States (On-site)$180k – $440k Yearly
5dXA
[Omni] Member of Technical Staff, World Model
xAI
Bay Area, California, United States (On-site)$180k – $440k Yearly
5dTE
1wCO
1wCO
Member of Technical Staff, Pretraining evaluations
Cohere
London, England, United Kingdom or Remote (Worldwide)
3wXA
Member of Technical Staff, Inference
xAI
Palo Alto, California, United States (On-site)$180k – $440k Yearly
4wGR
5dXA
Member of Technical Staff, RL Training Framework
xAI
Palo Alto, California, United States (On-site)$180k – $440k Yearly
3wNV
Senior Physics-Machine Learning Engineer - CAE
NVIDIA
Santa Clara, California, United States (On-site)$152k – $287.5k Yearly
5dAN
Performance Engineer, GPU
Anthropic
San Francisco, California, United States (Hybrid)$315k – $560k Yearly
3wXA
Member of Technical Staff, Model Evaluation
xAI
Palo Alto, California, United States (On-site)$180k – $440k Yearly
5dNV
Senior Software Engineer, Robotics - Isaac Lab
NVIDIA
Santa Clara, California, United States (On-site)$152k – $287.5k Yearly