Inference Jobs logo

Inference Jobs

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

PyTorch Jobs in United States

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

3d agoNV

Software Engineer, Machine Learning Inference - New College Grad 2026

NVIDIA

Santa Clara, California, United States (Hybrid)$108K – $195.5K Yearly
Machine Learning Engineering
Software Engineering
C++
Python
2mo agoNV

Senior GPU Functional Modeling Architect

NVIDIA

Santa Clara, California, United States (On-site)$152K – $287.5K Yearly
GPU Architecture
Computer Architecture
GPU Architecture
Computer Architecture
1w agoFI

Helix AI Engineer, Video Pretraining

Figure

San Jose, California, United States (On-site)
AI Engineer
Machine Learning Engineer
Python
PyTorch
2mo agoXA

Member of Technical Staff, Inference

xAI

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

Member of Technical Staff - Pre-Training

Reflection AI

San Francisco, California, United States (On-site)
AI Research
Machine Learning Engineer
Python
PyTorch
2w agoAN

Research Engineer, Pretraining Scaling

Anthropic

San Francisco, California, United States (On-site)$350K – $850K Yearly
Research Engineer
Machine Learning Engineer
JAX
TPU
3w agoTA

Senior Machine Learning Engineer, Voice AI

Together AI

San Francisco, California, United States (On-site)$200K – $260K Yearly
Machine Learning Engineer
AI Engineer
Machine Learning
Python
3mo agoOP

Research-Hardware Codesign Engineer

OpenAI

San Francisco, California, United States (Hybrid)$230K – $460K Yearly
Hardware Engineering
Machine Learning Research
Python
C++
1w agoFI

Helix AI Engineer, Pretraining

Figure

San Jose, California, United States (On-site)
AI Research Engineer
Machine Learning Engineer
Python
PyTorch
3w agoAN

Research Engineer, Performance RL

Anthropic

San Francisco, California, United States (Hybrid)$350K – $850K Yearly
AI Research & Engineering
Machine Learning Engineer
Reinforcement Learning
CUDA
2w agoD-

Principal Software ML Test Engineer

d-Matrix

Santa Clara, California, United States (Hybrid)$180K – $300K Yearly
Software Quality Assurance
Quality Engineering
Python
Pytest
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
7d 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
2mo agoNV

Senior System Software Engineer - Dynamo-Triton Inference Server

NVIDIA

Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
Software Engineering
Machine Learning Engineering
Deep Learning
C++
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
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
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 agoCE

Senior Full Stack LLM Engineer - Training

Cerebras

Sunnyvale, California, United States (On-site)
Machine Learning Engineer
LLM Engineering
Python
PyTorch