⇊ more filters ⇊

JAX Jobs

Browse 37 open JAX jobs from 4 companies including Nvidia, xAI, and Anthropic. Popular hiring locations include Santa Clara, Palo Alto, and Redmond.

Senior Software Engineer, CUDA Core Libraries at Nvidia logo Nvidia SENIOR
📍 Santa Clara, United States
💰 USD 184,000-356,500 per year
✅ On-site
AI @ 4 API @ 4 Algorithms @ 4 CUDA @ 4 Communication @ 4 GPU @ 4 HPC @ 4 JAX @ 7 LLVM @ 4 Performance Optimization @ 7 Profiling @ 4 PyTorch @ 7 Python @ 4 Rust @ 4
Senior Software Engineer, Spatial Intelligence And Foundation Models at Nvidia logo Nvidia SENIOR
📍 Santa Clara, United States
💰 USD 184,000-356,500 per year
✅ On-site
AI @ 4 Algorithms @ 4 Claude Code @ 6 Codex @ 6 Communication @ 4 Computer Vision @ 4 Data Pipelines @ 4 Deep Learning @ 4 Experimentation @ 6 GPU @ 4 JAX @ 4 PyTorch @ 4 Python @ 7 Robotics @ 4 TensorFlow @ 4
Senior Deep Learning Systems Architect at Nvidia logo Nvidia SENIOR
📍 Santa Clara, United States
💰 USD 224,000-356,500 per year
✅ On-site
AI @ 4 Algorithms @ 4 CUDA @ 1 Deep Learning @ 4 GPU @ 1 HPC @ 1 JAX @ 7 Machine Learning @ 4 OpenCL @ 1 Performance Analysis @ 4 PyTorch @ 7 Python @ 6 TensorFlow @ 7
Member Of Technical Staff - RL Training Framework at xAI logo xAI MIDDLE
📍 Palo Alto, United States
💰 USD 180,000-440,000 per year
✅ On-site
AI @ 3 Communication @ 6 Debugging @ 3 Distributed Systems @ 3 JAX @ 5 LLM @ 3 Leadership @ 3 Observability @ 3 Prioritization @ 6 Profiling @ 3 Python @ 5 Reinforcement Learning @ 6 Rust @ 5
Senior AI Compiler Engineer, Algorithms and Code-Generation at Nvidia logo Nvidia SENIOR
📍 Santa Clara, United States
💰 USD 152,000-241,500 per year
✅ On-site
AI @ 4 API @ 4 Algorithms @ 4 CUDA @ 7 Debugging @ 4 Deep Learning @ 4 GPU @ 6 JAX @ 4 LLVM @ 4 Mentoring @ 4 Performance Analysis @ 4 PyTorch @ 4 Python @ 4 Robotics @ 4
Senior Software Engineer, CUDA C++ Core Libraries at Nvidia logo Nvidia SENIOR
📍 Santa Clara, United States
💰 USD 184,000-356,500 per year
✅ On-site
AI @ 4 API @ 4 Algorithms @ 4 CUDA @ 4 Communication @ 4 GPU @ 4 HPC @ 4 Hiring @ 4 JAX @ 7 LLVM @ 4 Performance Optimization @ 7 Profiling @ 4 PyTorch @ 7 Python @ 4 Rust @ 4
Robotics Software Engineer - Robot Simulation and Benchmarking at Nvidia logo Nvidia MIDDLE
📍 Santa Clara, United States
💰 USD 152,000-241,500 per year
✅ On-site
AI @ 3 CUDA @ 3 Deep Learning @ 5 GPU @ 3 JAX @ 5 LLM @ 3 Machine Learning @ 3 PyTorch @ 5 Python @ 5 Reinforcement Learning @ 3 Robotics @ 3
Applied Research Intern, NLP - Fall 2026 at Nvidia logo Nvidia INTERN
📍 Santa Clara, United States
💸 USD 20-71 per hour
✅ On-site
AI @ 3 Algorithms @ 3 Deep Learning @ 6 GPU @ 3 JAX @ 3 NLP OSS @ 3 PyTorch @ 3 Python @ 3
Member of Technical Staff at xAI logo xAI MIDDLE
📍 Palo Alto, United States
💰 USD 324,000-396,000 per year
✅ On-site
AI @ 3 Communication @ 6 Data Science @ 3 Deep Learning @ 3 JAX @ 3 Java @ 3 LLM Machine Learning @ 3 Mentoring @ 3 Prioritization @ 6 Product Management @ 3 PyTorch @ 3 Python @ 3 Reinforcement Learning @ 3 TensorFlow @ 3
Software Engineering Manager, Robotics Neural Reconstruction And Real2Sim Applications at Nvidia logo Nvidia MIDDLE
📍 Santa Clara, United States
💰 USD 224,000-431,200 per year
✅ On-site
AI @ 3 Algorithms @ 3 Communication @ 3 Deep Learning @ 6 GPU @ 3 Hiring @ 3 JAX @ 3 Leadership @ 5 Machine Learning @ 5 PyTorch @ 3 Python @ 6 Reinforcement Learning @ 2 Robotics @ 3 Team Management @ 5 Technical Leadership @ 5 TensorFlow @ 3
Senior Deep Learning Performance Architect at Nvidia logo Nvidia SENIOR
📍 Santa Clara, United States
💰 USD 184,000-356,500 per year
✅ On-site
AI @ 4 Algorithms @ 4 Deep Learning @ 4 GPU @ 7 HPC JAX @ 4 LLM @ 3 Machine Learning @ 7 Performance Analysis @ 4 Profiling @ 4 PyTorch @ 4 Python @ 4 TensorRT @ 4
Senior Systems Software Engineer, AI Stack and Performance - DGX Station at Nvidia logo Nvidia SENIOR
📍 Santa Clara, United States
💰 USD 224,000-356,500 per year
✅ On-site
AI @ 4 Agentic AI @ 4 CUDA @ 4 Communication @ 4 Deep Learning @ 4 GPU @ 4 JAX @ 4 LLM @ 4 Machine Learning Microservices @ 4 NCCL @ 4 NVLink @ 4 Performance Analysis @ 8 Product Management @ 4 Profiling @ 4 PyTorch @ 4 Python @ 6 TensorFlow @ 4 TensorRT @ 4

Generation time: 0.134