JAX Jobs in California, United States

MLOps

Robotics Software Engineer

2mo agoNV

Senior Software Engineer - Robotics

NVIDIA

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

Senior Software Engineer

C++

3w agoET

Machine Learning Research Engineer

Etched

Cupertino, California, United States (On-site)

AI Research Scientist

Rust

3mo agoNV

Solutions Architect, AI Hyperscalers

NVIDIA

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

3mo agoPE

AI Engineer, Applied ML

Perplexity

San Francisco, California, United States (On-site)$210K – $385K Yearly

AI Engineer

Infrastructure Engineering

LLMs

3mo agoRA

Member of Technical Staff - GPU Infrastructure

Reflection AI

San Francisco, California, United States (On-site)

GPU Infrastructure

GPU Infrastructure

CUDA

2mo agoNV

Senior Compiler Engineer, AI Inference Platforms

NVIDIA

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

2mo agoNV

Senior Engineer - Deep Learning Compiler Verification and Infrastructure

NVIDIA

Santa Clara, California, United States (On-site)$140K – $224.3K Yearly

Compiler Engineering

Deep Learning

Compiler Verification

4w agoMA

AI Scientist - Palo Alto

Mistral AI

Palo Alto, California, United States (Hybrid)

2w agoAN

Performance Engineer, GPU

Anthropic

San Francisco, California, United States (Hybrid)$280K – $850K Yearly

4w agoNE

AI/ML Specialist Solution Architect (Early Talent)

Nebius

United States (Remote)

Product & Infrastructure

Solutions Architecture

Machine Learning Research Scientist

4w agoSC

Machine Learning Research Scientist / Engineer, Reasoning

Scale

San Francisco, California, United States (On-site)$252K – $315K Yearly

Natural Language Processing

3mo agoXA

Member of Technical Staff, Recommendation Systems

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly

Infrastructure Engineering

Recommendation Systems

Kafka

ClickHouse

3mo agoCO

Staff Software Engineer, GPU Infrastructure (HPC)

Cohere

Canada or Remote (Canada + 1 more)

Software Engineering

Kubernetes

GPU Clusters

1mo agoXA

Software Engineer, ML & Data Infra

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly

Machine Learning Infrastructure

Data Engineering

AI Research & Engineering

Rust

2mo agoNV

Senior Compiler Engineer, AI Inference Performance

NVIDIA

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

Compiler Engineering

AI Compiler Engineering

Compiler Engineering

Deep Learning

2mo agoNV

Senior Software Engineer - Deep Learning Compiler Verification and Infrastructure

NVIDIA

Santa Clara, California, United States (On-site)$140K – $224.3K Yearly

3w agoAN

Research Engineer, Performance RL

Anthropic

San Francisco, California, United States (Hybrid)$350K – $850K Yearly

Reinforcement Learning

CUDA

2mo agoSC

Machine Learning Research Engineer, GenAI Applied ML

Scale

San Francisco, California, United States (On-site)$176K – $220K Yearly

AI Research