Home
Jobs
United States
California
San Francisco
XLA

XLA Jobs in San Francisco, California, United States

Browse 16 XLA jobs in San Francisco, California, United States on Inference Jobs.

16 jobs

1 wk. ago

Research Engineer, Infrastructure, Training Systems

Thinking Machines Lab

San Francisco, California, United States (On-site)$350–475K Yearly

Machine Learning Infrastructure

Research Infrastructure (ML Infra)

PyTorch

JAX

2 wk. ago

Performance Engineer, GPU

Anthropic

San Francisco, California, United States (On-site)$280–850K Yearly

1 wk. ago

Research Engineer, Infrastructure, Inference

Thinking Machines Lab

San Francisco, California, United States (On-site)$350–475K Yearly

Machine Learning Infrastructure

AI Infrastructure

PyTorch

JAX

2 wk. ago

Research Engineer, Interpretability

Anthropic

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

AI Research

Machine Learning Engineer

Python

Rust

2 wk. ago

Staff+ Software Engineer, Inference Runtime

Anthropic

San Francisco, California, United States (Hybrid)$405–485K Yearly

Software Engineering - Infrastructure

Inference Runtime

Rust

Python

18 hr. ago Magic logo

Kernel Engineer

Magic

San Francisco, California, United States (On-site)$225–550K Yearly

Kernel Development

Machine Learning Infrastructure

NVIDIA Blackwell

Google TPUs

3 wk. ago

Member of Technical Staff, Training Infra Engineer

Cohere

Paris, FR or Remote (Worldwide)

Machine Learning Engineer

Infrastructure Engineering

Python

JAX

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

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)

7 days ago NVIDIA logo

Senior Deep Learning Compiler Engineer

NVIDIA

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

Deep Learning Compiler Engineer

Compiler Engineering

C/C++

Python

3 wk. ago

Member of Technical Staff, Modeling

Cohere

London, England, GB or Remote (Worldwide)

4 days ago Cohere logo

Member of Technical Staff, Agentic Environments

Cohere

North America, Europe, Middle East, and Africa (Remote)

Machine Learning Engineering

AI Infrastructure Engineering

Python

JAX

3 wk. ago

Member of Technical Staff, Search

Cohere

Worldwide (Remote)

7 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

7 days ago NVIDIA logo

Principal Deep Learning Communication Architect

NVIDIA

Santa Clara, California, US or Remote (United States)$272–431.3K Yearly

Deep Learning Architecture

HPC Engineering

NCCL

UCX

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