JAX/XLA Jobs
Explore JAX/XLA roles on Inference Jobs and apply today.
3w agoTM
Research Engineer, Infrastructure, Tinker
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K 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
2mo agoNV
Senior Engineer - Deep Learning Compiler Verification and Infrastructure
NVIDIA
Santa Clara, California, United States (On-site)$140K – $224.3K Yearly
4w agoNV
Senior Software Engineer, Profiling Services
NVIDIA
Tel Aviv-Yafo, Tel Aviv District, Israel (On-site)
3w agoTM
Research Engineer, Infrastructure, Inference
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
10h agoNV
Principal Deep Learning Communication Architect
NVIDIA
Santa Clara, California, United States (On-site)$272K – $431.3K Yearly
3w agoTM
Research Engineer, Infrastructure, Kernels
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
2mo agoNV
Senior AI Inference Compiler Engineer
NVIDIA
Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
2mo agoXA
Member of Technical Staff, Inference
xAI
Palo Alto, California, United States (On-site)$180K – $440K Yearly
1mo agoNV
Senior Compiler Tools Engineer - CUDA Tile
NVIDIA
Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
2mo agoCE
Python / PyTorch Developer — Frontend Inference Compiler – Dubai
Cerebras
United Arab Emirates (On-site)