JAX Jobs
Explore JAX roles on Inference Jobs and apply today.
3w agoTM
Research Engineer, Infrastructure, Numerics
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
4w agoSC
Machine Learning Research Scientist/ Engineer, Agents
Scale
San Francisco, California, United States (On-site)$275K – $350K Yearly
1mo agoRA
Member of Technical Staff - Research Software Engineer
Reflection AI
New York, New York, United States (On-site)
2w agoAN
Research Engineer, Pretraining Scaling
Anthropic
San Francisco, California, United States (On-site)$350K – $850K Yearly
4w agoXA
Member of Technical Staff - Multimodal
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
2mo agoDE
Staff Software Engineer, ML Infrastructure
Decagon
San Francisco, California, United States (On-site)$300K – $430K Yearly
2mo agoAI
ML Runtime Optimization Engineer - Lead
Applied Intuition
Sunnyvale, California, US$199.3K – $264.5K Yearly
3w agoTM
Research Engineer, Infrastructure, Inference
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
1d agoNV
Senior Solutions Architect, GPU - Cloud Service Providers
NVIDIA
Santa Clara, California, United States (On-site)$184K – $356.5K Yearly
5d agoTE
Sr. Software Engineer, AI Compiler
Tenstorrent
Toronto, Ontario, Canada (Hybrid)C$100K – C$500K Yearly
2mo agoXA
Member of Technical Staff, Model Evaluation
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
1mo agoAN
Research Engineer, Science of Scaling
Anthropic
London, England, United Kingdom (Hybrid)£260K – £630K Yearly
3w agoXT
AI Research Engineer
Xaira Therapeutics
Seattle, Washington, United States (On-site)$190K – $280K Yearly
2d agoNV
Software Engineering Intern, Robot Learning Platform - Summer 2026
NVIDIA
Toronto, Ontario, Canada (On-site)
3w agoTM
Software Engineer, Supercomputing
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
2mo agoNV
Senior Deep Learning Compiler Engineer - XLA
NVIDIA
Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
4w agoXA
Member of Technical Staff - Voice Model
xAI
Palo Alto, California, United States (On-site)$150K – $450K Yearly