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.

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
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
3mo agoD-

Machine Learning Research Intern

d-Matrix

Santa Clara, California, United States (Hybrid)$30 – $59 Hourly
Machine Learning Research
Machine Learning Engineer
PyTorch
Python
2mo agoPE

Research Engineering Manager - Model Training

Perplexity

San Francisco, California, United States (On-site)$300K – $470K Yearly
AI Research
Machine Learning
Python
PyTorch
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
2w agoAN

Research Engineer, Pre-training

Anthropic

San Francisco, California, United States (Hybrid)$350K – $850K Yearly
AI Research
Machine Learning Engineering
Python
PyTorch
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, Vision Expertise

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
AI Research
Computer Vision Research
Python
PyTorch
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
3mo agoCA

Researcher: Model Architecture

Cartesia

San Francisco, California, United States (On-site)$180K – $350K Yearly
Research
Machine Learning Research
State Space Models
PyTorch
3w agoTM

Research, Pre-Training Science

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Research Scientist
Machine Learning 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
3w agoTM

Research, Post-Training

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Research
Machine Learning Research
Python
PyTorch
4w agoAI

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
6d agoFI

Helix AI Engineer, Pretraining

Figure

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

Inference Engineering Manager

Perplexity

San Francisco, California, United States (On-site)$300K – $385K Yearly
AI
Engineering Management
Python
PyTorch
3mo agoCE

AI Models, Product Manager

Cerebras

Sunnyvale, California, United States or Remote (United States)
Product Management
AI Product Management
Python
PyTorch