PyTorch Jobs in California, United States

C/C++

1mo agoNV

Senior Performance Engineer - Deep Learning

NVIDIA

Santa Clara, California, United States (On-site)$152K – $241.5K Yearly

2w agoAN

Research Engineer, Pretraining Scaling

Anthropic

San Francisco, California, United States (On-site)$350K – $850K Yearly

Research Engineer

JAX

TPU

4w agoNE

AI/ML Specialist Solutions Architect

Nebius

United States + 1 more (Remote)$225K – $315K Yearly

Solutions Architecture

MLOps

Machine Learning

2mo agoCE

Senior Full Stack LLM Engineer - Training

Cerebras

Sunnyvale, California, United States (On-site)

LLM Engineering

Software Quality Assurance

PyTorch

1w agoD-

Principal Software ML Test Engineer

d-Matrix

Santa Clara, California, United States (Hybrid)$180K – $300K Yearly

Quality Engineering

Infrastructure Engineering

Pytest

3mo agoRA

Member of Technical Staff - GPU Infrastructure

Reflection AI

San Francisco, California, United States (On-site)

GPU Infrastructure

GPU Infrastructure

CUDA

3w agoTM

Research Engineer, Infrastructure, RL Systems

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly

Research Engineering

Reinforcement Learning

Distributed Training

3mo agoPE

AI Inference Engineer (San Francisco)

Perplexity

San Francisco, California, United States (On-site)$210K – $385K Yearly

AI Engineering

Rust

1mo agoNV

Senior Deep Learning Scientist, Multimodal Conversational AI

NVIDIA

Santa Clara, California, United States (On-site)$184K – $287.5K Yearly

Machine Learning Scientist

Large Language Models

3mo agoPE

AI Researcher

Perplexity

San Francisco, California, United States (On-site)$210K – $470K Yearly

AI Research

Large Language Models

3w agoTM

Research Engineer, Infrastructure, Kernels

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly

AI Research Engineer

CUDA

CuTe

4w agoFI

AI Training Infrastructure Engineer - Helix Team

Figure

San Jose, California, United States (On-site)$150K – $350K Yearly

MLOps

Multi-Modal Language Models

PyTorch

3mo agoBA

Software Engineer - Model Performance

Baseten

San Francisco, California, United States (On-site)$150K – $250K Yearly

2mo agoNV

Senior Research Scientist, Multi-Modal Language Models

NVIDIA

Santa Clara, California, United States (On-site)$192K – $356.5K Yearly

AI Research

Research Scientist

Computer Vision

3mo agoPL

Distributed Training Engineer

Periodic Labs

Menlo Park, California, United States (Hybrid)

Machine Learning Engineering

LLM Training

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

C++

3mo agoBA

Engineering Manager - Model Performance

Baseten

San Francisco, California, United States (On-site)$230K – $300K Yearly

Engineering Management

ML Engineering