JAX Jobs
Browse 128 JAX jobs on Inference Jobs.
41-60 of 128 jobs
2dNV
4dDE
Staff Software Engineer, ML Infrastructure
Decagon
San Francisco, California, United States (On-site)$300k – $430k Yearly
2wNV
Senior AI Software Engineer, GenAI Framework
NVIDIA
Santa Clara, California, United States (On-site)$152k – $287.5k Yearly
4wGR
7dAN
Software Engineer, Accelerator Build Infrastructure
Anthropic
San Francisco, California, United States (Hybrid)$315k – $405k Yearly
7dTM
Research Engineer, Infrastructure, Kernels
Thinking Machines Lab
San Francisco, California, United States (On-site)$350k – $475k Yearly
7dTM
Research Engineer, Infrastructure, Inference
Thinking Machines Lab
San Francisco, California, United States (On-site)$350k – $475k Yearly
7dTM
Research Engineer, Infrastructure, Training Systems
Thinking Machines Lab
San Francisco, California, United States (On-site)$350k – $475k Yearly
7dTM
Research Engineer, Infrastructure, Numerics
Thinking Machines Lab
San Francisco, California, United States (On-site)$350k – $475k Yearly
2dNV
Senior Deep Learning Compiler Engineer - XLA
NVIDIA
Santa Clara, California, United States (On-site)$152k – $241.5k Yearly
3dAN
Staff Software Engineer, Build Systems
Anthropic
London, England, United Kingdom (Hybrid)£325k – £390k Yearly
7dAN
Research Engineer, Pretraining Scaling
Anthropic
San Francisco, California, United States (On-site)$315k – $560k Yearly
7dXA
Member of Technical Staff, Recommendation Systems
xAI
Palo Alto, California, United States (On-site)$180k – $440k Yearly
7dNV
Senior Software Engineer - Storage
NVIDIA
Santa Clara, California, United States (On-site)$152k – $287.5k Yearly
7dTM
Software Engineer, Supercomputing
Thinking Machines Lab
San Francisco, California, United States (On-site)$350k – $475k Yearly