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

JAX/XLA Jobs

Browse 4 JAX/XLA jobs on Inference Jobs.

4 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
7d 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 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
Subscribe to this search

Get email updates when new jobs match this search.