Inference Jobs logo

Inference Jobs

  1. Home
  2. Jobs
  3. United States
  4. California
  5. XLA

XLA Jobs in California, United States

Discover XLA roles in California, United States on Inference Jobs and apply today.

2mo agoNV

Senior Deep Learning Compiler Engineer - XLA

NVIDIA

Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
Compiler Engineering
Deep Learning
C++
Compiler Optimization
3w agoTM

Research Engineer, Infrastructure, Training Systems

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Machine Learning Infrastructure
Research Infrastructure (ML Infra)
PyTorch
JAX
2mo agoXA

Member of Technical Staff, Midtraining

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Engineer
Data Engineer
Python
JAX
2mo agoNV

Senior AI Inference Compiler Engineer

NVIDIA

Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
Compiler Engineering
AI Engineering
Compiler Technologies
MLIR
4w agoXA

Member of Technical Staff - Multimodal

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Engineer
AI Research
Python
PyTorch
3mo agoMA

Kernel Engineer

Magic

San Francisco, California, United States (On-site)$225K – $550K Yearly
Kernel Development
Machine Learning Infrastructure
GPU Kernel Development
High Performance Computing
4w agoXA

Member of Technical Staff - Imagine Model

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Engineer
AI Researcher
Multimodal Machine Learning
Python
2mo agoXA

Member of Technical Staff, Model Evaluation

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Engineer
Model Evaluation
Python
JAX
2w agoAN

Research Engineer, Interpretability

Anthropic

San Francisco, California, United States (Hybrid)$315K – $560K Yearly
AI Research
Machine Learning Engineer
Python
Rust
2mo agoNV

Senior Compiler Engineer, AI Inference Platforms

NVIDIA

Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
Compiler Engineering
AI Engineering
C++
Python
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
3mo agoCO

Member of Technical Staff, Modeling

Cohere

London, England, United Kingdom or Remote (Worldwide)
Modeling
AI Research
Python
TensorFlow
4w agoAI

ML Runtime Optimization Engineer

Applied Intuition

Sunnyvale, California, United States (On-site)$159.1K – $199.3K Yearly
Machine Learning Engineering
Embedded Software Engineering
PyTorch
JAX
3w agoTM

Research Engineer, Infrastructure, Numerics

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Research Infrastructure (ML Infra)
Machine Learning Infrastructure
PyTorch
JAX
1mo agoXA

Software Engineer, ML & Data Infra

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Infrastructure
Data Engineering
Python
Rust
2w agoAN

Performance Engineer, GPU

Anthropic

San Francisco, California, United States (Hybrid)$280K – $850K Yearly
Performance Engineer
GPU Engineer
CUDA
GPU Programming
3w agoTM

Research Engineer, Infrastructure, Tinker

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
ML Infrastructure
Machine Learning Engineer
Deep Learning Frameworks
PyTorch