JAX/XLA Jobs
Explore JAX/XLA roles on Inference Jobs and apply today.
2mo agoNV
Senior Deep Learning Compiler Engineer - XLA
NVIDIA
Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
2mo agoXA
Member of Technical Staff, Midtraining
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
3w agoTM
Research Engineer, Infrastructure, Training Systems
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
3mo agoCO
4w agoXA
Member of Technical Staff - Multimodal
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
3mo agoCO
Member of Technical Staff, Training Performance Engineer
Cohere
London, England, United Kingdom (Hybrid)
3mo agoCO
2mo agoXA
Member of Technical Staff, Model Evaluation
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
3w agoXA
Member of Technical Staff - Imagine Model
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
3mo agoCO
3mo agoCO
Member of Technical Staff, Integration/RL Team (Research Engineer)
Cohere
Paris, Paris, France or Remote (United States + 3 more)
3mo agoCO
1w agoAN
Research Engineer, Interpretability
Anthropic
San Francisco, California, United States (Hybrid)$315K – $560K Yearly
3w agoTM
Research Engineer, Infrastructure, Numerics
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
2mo agoAI
ML Runtime Optimization Engineer - Lead
Applied Intuition
Sunnyvale, California, US$199.3K – $264.5K Yearly
4w agoAI
ML Runtime Optimization Engineer
Applied Intuition
Sunnyvale, California, United States (On-site)$159.1K – $199.3K Yearly
2mo agoNV
Senior Compiler Engineer, AI Inference Platforms
NVIDIA
Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
1w agoAN
Performance Engineer, GPU
Anthropic
San Francisco, California, United States (Hybrid)$280K – $850K Yearly