Inference Jobs logo

Inference Jobs

  1. Home
  2. Jobs
  3. United States
  4. California
  5. Distributed Training

Distributed Training Jobs in California, United States

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

3w agoTM

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
4w agoVA

GPU Systems Engineer – HPC / Parallel Computing

Vast.ai

San Francisco, California, United States (On-site)$160K – $320K Yearly
GPU Computing
HPC Engineering
HPC
Parallel Computing
3mo agoNV

Solutions Architect, AI Cloud Partner Performance

NVIDIA

Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
Solutions Architect
Cloud Computing
AWS
Azure
3mo agoOP

Software Engineer, Distributed Data Systems (Sora)

OpenAI

San Francisco, California, United States (Hybrid)$255K – $405K Yearly
Software Engineering
Data Engineering
Distributed Systems
Data Infrastructure
2mo agoNV

Senior ML Framework Performance Engineer - AI for Science at Scale

NVIDIA

Santa Clara, California, United States (On-site)$184K – $287.5K Yearly
Machine Learning Engineer
Performance Engineer
PyTorch
JAX
2w agoAN

Research Engineer, Pretraining Scaling

Anthropic

San Francisco, California, United States (On-site)$350K – $850K Yearly
Research Engineer
Machine Learning Engineer
JAX
TPU
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
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
1d agoNV

Principal Deep Learning Communication Architect

NVIDIA

Santa Clara, California, United States (On-site)$272K – $431.3K Yearly
Deep Learning Architecture
HPC Engineering
NCCL
UCX
4w agoTA

AI Researcher, Core ML

Together AI

San Francisco, California, United States (On-site)$200K – $280K Yearly
AI Research
Core ML
SGLang
vLLM
2mo agoTA

Machine Learning, Platform Engineer

Together AI

San Francisco, California, United States (On-site)$160K – $250K Yearly
Engineering
Platform Engineering
Machine Learning
Platform Engineering
3mo agoOP

Software Engineer, Accelerators

OpenAI

San Francisco, California, United States (On-site)$310K – $380K Yearly
Machine Learning Infrastructure
Software Engineering
Kernels
AI Infrastructure
3mo agoPO

Member of Engineering (Scalability)

Poolside

United Kingdom or Remote (Europe, Middle East, and Africa, North America)
R&D
Machine Learning Infrastructure
Linux Kernel
PyTorch
3d agoSC

Tech Lead Manager- MLRE, ML Systems

Scale

San Francisco, California, United States (On-site)$264.8K – $331K Yearly
Machine Learning
MLOps
LLM
Distributed Systems
3mo agoPL

LLM Inference Engineer

Periodic Labs

Menlo Park, California, United States (On-site)
LLM Engineering
Machine Learning Infrastructure
LLM Inference
TensorRT-LLM
4w agoOP

TL, Research Inference

OpenAI

San Francisco, California, United States (On-site)$380K – $555K Yearly
Research Engineering
Machine Learning Infrastructure
High-Performance Inference
Model Execution
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
3mo agoOP

Software Engineer, Collective Communication

OpenAI

San Francisco, California, United States (Hybrid)$380K – $555K Yearly
Software Engineering
HPC Engineer
C++
CUDA