Home
Jobs
United States
California
Cupertino
JAX

JAX Jobs in Cupertino, California, United States

Browse 39 JAX jobs in Cupertino, California, United States on Inference Jobs.

39 jobs

3 wk. ago

Machine Learning Research Engineer

Etched

Cupertino, California, United States (On-site)

Machine Learning Engineer

AI Research Scientist

Python

Rust

6 days ago NVIDIA logo

Senior Deep Learning Framework Communications Engineer

NVIDIA

United States (Remote)$184–287.5K Yearly

Deep Learning Engineer

HPC Engineer

PyTorch

JAX

6 days ago NVIDIA logo

Senior Software Engineer, AI and DL Kernel Libraries

NVIDIA

United States (Remote)$184–287.5K Yearly

AI Systems Engineering

GPU Computing

Python

C++

6 days ago NVIDIA logo

Senior Deep Learning Frameworks CUDA Software Engineer

NVIDIA

Santa Clara, California, US or Remote (United States)$224–356.5K Yearly

Deep Learning Frameworks Engineering

CUDA Software Engineering

CUDA

PyTorch

3 wk. ago

Senior Software Engineer, Matrix Multiplication

NVIDIA

United States (Remote)$184–287.5K Yearly

AI Systems Engineering

GPU Kernel Development

CUDA C/C++

cuTile

2 days ago Nebius logo

ML Infrastructure Engineer

Nebius

Europe + 1 more (Remote)

ML Infrastructure Engineering

GPU Engineering

GPU

CUDA

3 wk. ago

Member of Technical Staff (ML)

Reka

Worldwide (Remote)

Machine Learning Engineer

AI Research

Deep Learning

Machine Learning

6 days ago NVIDIA logo

Senior DGX Cloud AI Infrastructure Software Engineer

NVIDIA

Worldwide (Remote)$184–356.5K Yearly

AI Infrastructure Engineer

Software Engineer

Python

C/C++

3 wk. ago

Member of Technical Staff, Training Infra Engineer

Cohere

Paris, FR or Remote (Worldwide)

Machine Learning Engineer

Infrastructure Engineering

Python

JAX

6 days ago NVIDIA logo

Developer Advocate – Reinforcement Learning

NVIDIA

Santa Clara, California, US or Remote (United States)$152–287.5K Yearly

Developer Advocate

Developer Relations

Reinforcement Learning

LLM

4 wk. ago

Senior Compiler Engineer, AI Inference Platforms

NVIDIA

Santa Clara, California, US or Remote (United States)$152–241.5K Yearly

3 wk. ago

Staff Software Engineer, GPU Infrastructure (HPC)

Cohere

CA or Remote (Canada + 1 more)

Software Engineering

Infrastructure Engineering

Kubernetes

GPU Clusters

3 wk. ago

Senior ML Systems Engineer, Frameworks & Tooling

Cohere

London, England, GB or Remote (Worldwide)

Machine Learning Engineer

MLOps

JAX

Distributed Systems

3 wk. ago

Member of Technical Staff, Training Performance Engineer

Cohere

London, England, GB or Remote (Worldwide)

Machine Learning Engineer

Performance Engineer

Python

JAX

3 wk. ago

Machine Learning Intern/Co-op (Fall, 2026)

Cohere

Canada + 3 more (Remote)

6 days ago NVIDIA logo

Senior Solutions Architect, Financial Services Capital Markets

NVIDIA

United States (Remote)$224–356.5K Yearly

Solutions Architecture

Financial Services Technology

Python

C++

2 days ago Baseten logo

Post-Training Research Engineer

Baseten

United States (Remote)$200–275K Yearly

Machine Learning Engineer

Research Engineer

PyTorch

TensorFlow

3 wk. ago

Open-Source Machine Learning Engineer - US Remote

Hugging Face

United States (Remote)

Open Source

Machine Learning Engineering

Python

PyTorch

3 wk. ago

Member of Technical Staff, Modeling

Cohere

London, England, GB or Remote (Worldwide)

3 wk. ago

Research Scientist, Generalist Embodied Agent Research - PhD New College Grad 2026

NVIDIA

United States (Remote)$168–304.8K Yearly

Robotics Research Scientist

Embodied AI Research

PyTorch

JAX

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