Inference Jobs logo

Inference Jobs

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

PyTorch Jobs in California, United States

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

2mo agoNV

Senior Software Engineer - Deep Learning Compiler Verification and Infrastructure

NVIDIA

Santa Clara, California, United States (On-site)$140K – $224.3K Yearly
Software Engineering
DevOps
CI/CD
Deep Learning
2mo agoTA

Research Engineer, Frontier Speculative Decoding

Together AI

San Francisco, California, United States (On-site)$190K – $270K Yearly
Machine Learning Research
Research Engineer
Speculative Decoding
Data Curation
3mo agoOP

Training Performance Engineer

OpenAI

San Francisco, California, United States (Hybrid)$250K – $460K Yearly
Performance Engineering
Machine Learning Engineering
Python
C++
3w agoCO

Staff Engineer - Perf and Benchmarking

CoreWeave

Sunnyvale, California, United States (Hybrid)$206K – $333K Yearly
Performance Engineering
Machine Learning Engineering
MLPerf
Kubernetes
1w agoGR

AI Performance Engineer

Graphcore

Milpitas, California, United States (On-site)
Performance 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
1mo agoNV

Senior DL Algorithms Engineer - Inference Performance

NVIDIA

Santa Clara, California, United States (On-site)$184K – $356.5K Yearly
Deep Learning Engineer
Algorithms Engineer
Deep Learning
Inference
1mo agoNV

Senior Deep Learning Algorithm Engineer, Training Framework

NVIDIA

Santa Clara, California, United States (Hybrid)$184K – $287.5K Yearly
Deep Learning
AI Infrastructure
Megatron Core
NeMo Framework
2mo agoNV

Principal Product Manager, AI Frameworks

NVIDIA

Santa Clara, California, United States (On-site)$240K – $379.5K Yearly
Product Management
AI Product Management
AI Frameworks
Post-training
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
4w agoTA

LLM Inference Frameworks and Optimization Engineer

Together AI

San Francisco, California, United States (On-site)$160K – $230K Yearly
Machine Learning Engineer
AI Research Engineer
LLM Inference
TensorRT-LLM
3mo agoBA

GPU Kernel Engineer

Baseten

San Francisco, California, United States (On-site)$185K – $250K Yearly
GPU Kernel Engineering
AI Infrastructure
CUDA
C++
2w agoAN

TPU Kernel Engineer

Anthropic

San Francisco, California, United States (Hybrid)$280K – $850K Yearly
Kernel Engineering
Machine Learning Engineering
TPU
ML Systems
3mo agoPL

LLM Inference Engineer

Periodic Labs

Menlo Park, California, United States (On-site)
LLM Engineering
Machine Learning Infrastructure
LLM Inference
TensorRT-LLM
2mo agoNV

Senior Applied Deep Learning Research Scientist, Efficiency

NVIDIA

Santa Clara, California, United States (On-site)$192K – $356.5K Yearly
Deep Learning Research
Applied Science
Deep Learning
Python
3mo agoBA

Senior Software Engineer - Model Training

Baseten

San Francisco, California, United States (On-site)$200K – $275K Yearly
Engineering
Model Training
ML Infrastructure
Model Training
4w agoTA

Systems Research Engineer, GPU Programming

Together AI

San Francisco, California, United States (Hybrid)$160K – $230K Yearly
AI Research
Machine Learning Engineering
GPU Programming
CUDA
3mo agoOP

Training: ML Framework Engineer

OpenAI

San Francisco, California, United States (Hybrid)$245K – $385K Yearly
Machine Learning Engineer
ML Infrastructure
Python
Machine Learning
1mo agoMA

Member of Technical Staff, Pre-training Systems

Magic

San Francisco, California, United States (On-site)$225K – $550K Yearly
Machine Learning Infrastructure
Distributed Systems
Distributed Systems
Deep Learning