1

Senior Pytorch Jobs in California (NOW HIRING)

Senior Compiler Engineer

San Francisco, CA · On-site

$123K - $169K/yr

They are seeking a Senior Compiler Engineer to design and build core compiler infrastructure ... Luminal provides a machine learning compiler and serverless cloud platform that automates PyTorch ...

Senior ML Software Engineer, Data Plane

Cupertino, CA · On-site

$151K - $199K/yr

... PyTorch model definition through distributed execution on custom hardware. - Integrate custom accelerator backends into open-source ML serving frameworks (vLLM, PyTorch), including scheduler ...

Senior ML Software Engineer, Data Plane

Cupertino, CA · On-site

$151K - $199K/yr

... PyTorch model definition through distributed execution on custom hardware. - Integrate custom accelerator backends into open-source ML serving frameworks (vLLM, PyTorch), including scheduler ...

Senior AI Engineer

San Francisco, CA · On-site

$123K - $169K/yr

We're looking for a Senior AI engineer to build the core intelligence behind AI teammates for ... Strong Python skills with frameworks like PyTorch or TensorFlow * Familiarity with AI agent ...

Senior Software Engineer, NCCL

Santa Clara, CA · On-site

$143K - $189K/yr

They are seeking a highly motivated senior software engineer for their communication libraries and ... NCCL for TensorFlow/Pytorch) and HPC programming interfaces (e.g. UCX for MPI/OpenSHMEM) on GPU ...

They are looking for a Senior AI/ML Engineer to own model training pipelines, evaluation systems ... Strong Python and PyTorch (or JAX) fundamentals * Experience with distributed training, GPU ...

next page

Showing results 1-20

Senior Pytorch information

What are the most commonly searched types of Pytorch jobs in California?

The most popular types of Pytorch jobs in California are:

Infographic showing various Senior Pytorch job openings in California as of August 2026, with employment types broken down into 1% Internship, 85% Full Time, 10% Part Time, and 4% Contract. Highlights an 79% Physical, 3% Hybrid, and 18% Remote job distribution.

Senior ML Infrastructure Engineer (PyTorch, Kubernetes, GPU Training)

Finoit Inc.

Redwood City, CA • On-site

$132K - $180K/yr

Other

Re-posted 10 days ago


Job description

Senior ML Infrastructure Engineer (PyTorch, Kubernetes, GPU Training)

Short Job Description

We are seeking a Senior ML Infrastructure Engineer to design and scale the infrastructure powering large-scale machine learning training workloads. In this role, you'll build high-performance GPU training platforms, optimize distributed training pipelines, and improve the developer experience for ML researchers.

Responsibilities:

  • Design and scale distributed ML training infrastructure for large GPU clusters.
  • Build and optimize training pipelines using PyTorch, DeepSpeed, and distributed training frameworks.
  • Develop and maintain job scheduling systems using Kubernetes and/or SLURM.
  • Create high-throughput data pipelines for large-scale multimodal datasets.
  • Optimize GPU utilization, memory efficiency, and overall system performance.
  • Build low-latency inference pipelines for production ML deployments.

Required Skills:

  • 7+ years of experience in ML Infrastructure, HPC, or Distributed Systems.
  • Strong experience with PyTorch, DeepSpeed, FSDP, ZeRO, or similar distributed training frameworks.
  • Hands-on experience with Kubernetes, cloud platforms (AWS/Google Cloud Platform), and containerized environments.
  • Strong understanding of distributed systems, GPU optimization, NCCL, memory management, and performance tuning.
  • Experience building scalable ML infrastructure from development through production.

Location: Redwood City, CA (On-site)
Employment Type: Full-Time

Nice to Have:

  • Experience with multimodal AI, robotics data pipelines, Triton, TensorRT, custom ML kernels, or ML compiler/runtime optimization.