Home
Jobs
United States
California
Santa Clara
JAX

JAX Jobs in Santa Clara, California, United States

Browse 69 JAX jobs in Santa Clara, California, United States on Inference Jobs.

69 jobs

1 wk. ago

Senior Manager, Software Engineering - JAX

NVIDIA

Santa Clara, California, United States (On-site)$272–431.3K Yearly

Software Engineering Management

AI Infrastructure

JAX

XLA

4 wk. ago

Senior Software Engineer, Deep Learning - MLIR TRT

NVIDIA

Santa Clara, California, United States (On-site)$184–356.5K Yearly

4 wk. ago

Product Marketing Manager - AI Platform Software

NVIDIA

Santa Clara, California, United States (On-site)$184–356.5K Yearly

Product Marketing

Technical Product Marketing

PyTorch

JAX

1 wk. ago

Senior Deep Learning Tools Engineer – CUDA Tile

NVIDIA

Santa Clara, California, United States (Hybrid)$152–241.5K Yearly

4 wk. ago

Senior HPC Performance Engineer - AI for Science at Scale

NVIDIA

Santa Clara, California, United States (On-site)$184–287.5K Yearly

HPC Engineering

Performance Engineering

C++

Python

4 wk. ago

Senior Research Engineer - Autonomous Vehicles

NVIDIA

Santa Clara, California, United States (On-site)$184–356.5K Yearly

Research Engineering

Machine Learning Research

Deep Learning

Reinforcement Learning

4 wk. ago

Senior Research Engineer, Foundation Model Training Infrastructure

NVIDIA

Santa Clara, California, United States (On-site)$224–356.5K Yearly

AI Infrastructure

Machine Learning Engineering

MLOps

AI Infrastructure

3 wk. ago

Senior Solutions Architect, GPU Performance and LLM - Cloud Service Providers

NVIDIA

Santa Clara, California, United States (On-site)$184–356.5K Yearly

Solutions Architecture

GPU Performance Engineering

AI/ML

HPC

4 wk. ago

Technical Marketing Engineer - AI Platform Software

NVIDIA

Santa Clara, California, United States (Hybrid)$136–253K Yearly

Technical Marketing Engineer

Technical Marketing

Python

C/C++

4 wk. ago

Senior Performance Engineer - Deep Learning

NVIDIA

Santa Clara, California, United States (On-site)$152–241.5K Yearly

1 wk. ago

Senior Software Engineer — cuEquivariance

NVIDIA

Santa Clara, California, United States (On-site)$224–356.5K Yearly

GPU Kernel Engineering

Senior Software Engineering

CUDA

C++

1 wk. ago

Senior Software Engineer, CUTLASS Performance

NVIDIA

Santa Clara, California, United States (On-site)$152–287.5K Yearly

Software Engineering

Performance Engineering

Python

C++

4 wk. ago

Principal Software Engineer, Profiling Services

NVIDIA

Santa Clara, California, United States (On-site)$272–431.3K Yearly

1 wk. ago

Senior High-Performance LLM Training Engineer

NVIDIA

Santa Clara, California, United States (Hybrid)$184–356.5K Yearly

LLM Training Engineering

Deep Learning Engineering

PyTorch

JAX

3 days ago Tenstorrent logo

Sr. Engineer, Software - AI Compiler

Tenstorrent

Santa Clara, California, United States (Hybrid)$100–500K Yearly

3 wk. ago

Engineering Manager, AI Compiler Analysis

NVIDIA

Santa Clara, California, United States (On-site)$168–322K Yearly

Engineering Management

Compiler Engineering

MLIR

TensorRT

3 days ago NVIDIA logo

Solutions Architect, LLM Model Builder

NVIDIA

Santa Clara, California, United States (On-site)$152–241.5K Yearly

Solutions Architecture

Machine Learning Engineering

LLMs

VLMs

4 wk. ago

Senior Inference Technical Product Marketing Manager - Accelerated Computing

NVIDIA

Santa Clara, California, United States (On-site)$148–287.5K Yearly

Technical Product Marketing

AI Product Marketing

Accelerated Computing

AI Inference

1 wk. ago

Software Engineering Manager, Robotics Neural Reconstruction and Real2Sim Applications

NVIDIA

Santa Clara, California, United States (On-site)$224–431.3K Yearly

Engineering Management

Robotics Engineering

Python

C++

4 days ago NVIDIA logo

Senior Manager, Software Development - GPU Accelerated Storage

NVIDIA

Santa Clara, California, United States (On-site)$272–488.8K Yearly

Software Development Management

GPU Computing

CUDA

Jobs on Inference Jobs

Browse jobs on Inference Jobs.

JobsAI-ranked

Inference Jobs

Discover the latest AI roles from Inference Jobs.

Powered byCavuno

For Candidates

Jobs
Companies

For Companies

Post a job
Pricing

Resources

Locations
Salaries
Sitemap

About

About