JAX Jobs in California, United States
Discover JAX roles in California, United States on Inference Jobs and apply today.
1mo agoNV
Senior Deep Learning Algorithm Engineer, Training Framework
NVIDIA
Santa Clara, California, United States (Hybrid)$184K – $287.5K Yearly
1mo agoNV
Senior HPC Performance Engineer - AI for Science at Scale
NVIDIA
Santa Clara, California, United States (On-site)$184K – $287.5K Yearly
2mo agoNV
Senior Software Engineer - Storage
NVIDIA
Santa Clara, California, United States (On-site)$152K – $287.5K Yearly
3mo agoOP
Research-Hardware Codesign Engineer
OpenAI
San Francisco, California, United States (Hybrid)$230K – $460K Yearly
1mo agoNV
AI Inference Performance Engineer - New College Grad 2026
NVIDIA
Santa Clara, California, United States (On-site)$124K – $241.5K Yearly
3mo agoOP
Training Performance Engineer
OpenAI
San Francisco, California, United States (Hybrid)$250K – $460K Yearly
2mo agoNV
Research Scientist, Quantum Computing and AI - New College Grad 2026
NVIDIA
Santa Clara, California, United States (On-site)$168K – $264.5K Yearly
2w agoXT
BioMedical AI Research Engineer
Xaira Therapeutics
South San Francisco, California, United States (On-site)$190K – $280K Yearly
2mo agoNV
High-Performance LLM Training Engineer - New College Grad 2026
NVIDIA
Santa Clara, California, United States (On-site)$124K – $195.5K Yearly
2mo agoNV
Deep Learning Performance Architect - New College Graduate 2026
NVIDIA
Santa Clara, California, United States (On-site)$124K – $241.5K Yearly
1d agoNV
Principal Deep Learning Communication Architect
NVIDIA
Santa Clara, California, United States (On-site)$272K – $431.3K Yearly
2w agoNV
NVIDIA 2026 Internships: Deep Learning - US
NVIDIA
Santa Clara, California, United States (On-site)$20 – $71 Hourly
3mo agoLA
JavaScript Engineer (Open Source Team)
LangChain
San Francisco, California, United States (On-site)$150K – $225K Yearly
2w agoTA
Senior Platform Engineer, Voice AI
Together AI
San Francisco, California, United States (On-site)$200K – $260K Yearly
3mo agoNV
Senior Developer Technology Engineer - Windows AI Platform
NVIDIA
Santa Clara, California, United States (On-site)$184K – $356.5K Yearly
1w agoPR
AI Trainer - Advanced JavaScript Developers (US & Canada)
Prolific
United States + 1 more (Remote)From $40 Hourly
1mo agoNV
1w agoPR
AI Trainer - Advanced Java Developers (US & Canada)
Prolific
United States + 1 more (Remote)$40 – $40 Hourly