JAX Jobs
Explore JAX roles on Inference Jobs and apply today.
3mo agoOP
Software Engineer, Inference – AMD GPU Enablement
OpenAI
San Francisco, California, United States (On-site)$325K – $490K Yearly
2mo agoNV
Senior Software Engineer - Deep Learning Compiler Verification and Infrastructure
NVIDIA
Santa Clara, California, United States (On-site)$140K – $224.3K Yearly
4w agoTE
Sr. Engineer, Software - AI Compiler
Tenstorrent
Santa Clara, California, United States (Hybrid)$100K – $500K Yearly
3w agoAN
Research Engineer, Performance RL
Anthropic
San Francisco, California, United States (Hybrid)$350K – $850K Yearly
2mo agoNV
Senior Software Engineer - Robotics
NVIDIA
Santa Clara, California, United States (On-site)$184K – $356.5K Yearly
3mo agoLA
Senior Frontend Engineer, Observability & Evals Platform (LangSmith)
LangChain
San Francisco, California, United States (On-site)$175K – $225K Yearly
2w agoAN
Research Engineer, Interpretability
Anthropic
San Francisco, California, United States (Hybrid)$315K – $560K Yearly
2mo agoTM
Research Infrastructure Engineer, Research Acceleration
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
2mo agoNV
Senior AI Applications Engineer - Finance
NVIDIA
Santa Clara, California, United States (On-site)$168K – $264.5K Yearly
2mo agoSC
Machine Learning Research Engineer, GenAI Applied ML
Scale
San Francisco, California, United States (On-site)$176K – $220K Yearly
2mo agoNV
Senior AI Developer Technology Engineer, Financial Sector
NVIDIA
Santa Clara, California, United States (Hybrid)$152K – $241.5K Yearly
3mo agoOP
Research-Hardware Codesign Engineer
OpenAI
San Francisco, California, United States (Hybrid)$230K – $460K Yearly
3w agoTM
Research, Pre-Training Data
Thinking Machines Lab
San Francisco, California, United States (On-site)$350K – $475K Yearly
2mo agoNV
Senior Engineer - Deep Learning Compiler Verification and Infrastructure
NVIDIA
Santa Clara, California, United States (On-site)$140K – $224.3K Yearly
3mo agoLA
JavaScript Engineer (Open Source Team)
LangChain
San Francisco, California, United States (On-site)$150K – $225K Yearly