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.

1mo agoXA

Member of Technical Staff - Voice Model

xAI

Palo Alto, California, United States (On-site)$150K – $450K Yearly
Machine Learning Engineer
AI Engineer
Python
Spark
3w agoTM

Research Engineer, Infrastructure, Inference

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Machine Learning Infrastructure
AI Infrastructure
PyTorch
JAX
3mo agoCO

Senior Member of Technical Staff, Multimodal AI

Cohere

San Francisco, California, United States or Remote (Worldwide)
Machine Learning Engineer
AI Research Engineer
Python
JAX
3mo agoBA

Engineering Manager - Model Performance

Baseten

San Francisco, California, United States (On-site)$230K – $300K Yearly
Engineering Management
ML Engineering
Python
C++
2mo agoNV

Senior AI Inference Compiler Engineer

NVIDIA

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

Software Engineer, Inference AI/ML

CoreWeave

Sunnyvale, California, United States (Hybrid)$92K – $135K Yearly
Software Engineering
Machine Learning Engineering
Python
Go
3mo agoBA

Senior Product Engineer - Training Platform

Baseten

San Francisco, California, United States (On-site)$200K – $275K Yearly
Machine Learning Infrastructure
Distributed Systems Engineering
Kubernetes
PyTorch
3w agoTM

Research, Post-Training

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Research
Machine Learning Research
Python
PyTorch
21h agoXA

Member of Technical Staff - Multimodal Understanding

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Engineering
Multimodal AI
Python
PyTorch
2w agoAN

Research Engineer, Interpretability

Anthropic

San Francisco, California, United States (Hybrid)$315K – $560K Yearly
AI Research
Machine Learning Engineer
Python
Rust
1w agoFI

Helix AI Engineer, Modeling

Figure

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

Research Engineer, Machine Learning (Horizons)

Anthropic

San Francisco, California, United States (Hybrid)$500K – $850K Yearly
Machine Learning Engineering
AI Research
Python
Reinforcement Learning
3mo agoOP

Software Engineer, Accelerators

OpenAI

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

Neural Graphics Engineer

NVIDIA

Santa Clara, California, United States (On-site)$124K – $241.5K Yearly
Graphics Engineer
Software Engineer
C++
Python
3mo agoNV

Senior Software Engineer, Metropolis Vision AI

NVIDIA

Santa Clara, California, United States (On-site)$152K – $241.5K Yearly
Software Engineering
Computer Vision
Modern C++
Python
1mo 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, Audio Expertise

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Machine Learning Research
AI Research
Python
PyTorch
3mo agoPE

AI Researcher

Perplexity

San Francisco, California, United States (On-site)$210K – $470K Yearly
AI
AI Research
Large Language Models
Deep Learning