Inference Jobs logo
Inference Jobs
  1. Home
  2. Jobs
  3. PyTorch/XLA

PyTorch/XLA Jobs

Browse 11 PyTorch/XLA jobs on Inference Jobs.

11 jobs
2w agoNVIDIA logoNV

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
2w agoThinking Machines Lab logoTM

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
2w agoThinking Machines Lab logoTM

Research Engineer, Infrastructure, Kernels

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Machine Learning Infrastructure
AI Research Engineer
CUDA
CuTe
5d agoApplied Intuition logoAI

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
2w agoNVIDIA logoNV

Senior AI Inference Compiler Engineer

NVIDIA

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

Performance Engineer, GPU

Anthropic

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

Research Engineer

Graphcore

Bristol, England, United Kingdom (On-site)
Machine Learning Research
AI Research
Python
PyTorch
2w agoGraphcore logoGR

Pytorch Engineer

Graphcore

Bristol, England, United Kingdom (On-site)
Software Engineering
Machine Learning
PyTorch
Python
Subscribe to this search

Get email updates when new jobs match this search.