ZipRecruiter

Log In

1

Senior Pytorch Jobs in California (NOW HIRING)

Luminal

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 ...

Nvidia

Senior Software Engineer, AI Networking

Santa Clara, CA · On-site

$143K - $189K/yr

Proven ability to apply GNNs/transformers-based optimization to PyTorch model graph and Kineto ... If you are a senior data engineer passionate about building largescale, highimpact data platforms ...

Numos AI

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 ...

NVIDIA

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 ...

Hum AI

Senior Machine Learning Engineer

San Francisco, CA · On-site

$123K - $169K/yr

Senior Machine Learning Engineer Location: San Francisco About Hum.ai Hum.ai is building planetary ... Proficiency with Python, Ray Trainer, PyTorch, and Anyscale framework. * Familiarity with cloud ...

next page

Showing results 1-20

All JobsPytorch JobsSenior Pytorch JobsSenior Pytorch Jobs in California

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:
What are popular job titles related to Senior Pytorch jobs in California? For Senior Pytorch jobs in California, the most frequently searched job titles are:
What job categories do people searching Senior Pytorch jobs in California look for? The top searched job categories for Senior Pytorch jobs in California are:
Senior Pytorch jobs near you
Infographic showing various Senior Pytorch job openings in California as of August 2026, with employment types broken down into 1% Internship, 91% Full Time, 4% Part Time, and 4% Contract. Highlights an 76% Physical, 2% Hybrid, and 22% Remote job distribution.

Senior Machine Learning Research Engineer, AIML

Apple

Cupertino, CA • On-site

$128K - $177K/yr

Full-time

Re-posted 4 days ago

Apple rating

8.0

Company rating: 8.0 out of 10

Based on 677 frontline employees who took The Breakroom Quiz

7th of 30 rated technology retailers

Job description

Are you passionate about Generative AI and excited to work on groundbreaking modeling technologies that will enrich the lives of billions? The AIML team within Apple's software organization is a multidisciplinary group operating at the intersection of Multimodal Foundation Models, Efficient and Scalable ML Infrastructure, and Personalized Intelligent Experiences. As a senior machine learning engineer on our team, you will design software systems and algorithms that enable performant, scalable training and inference for Apple's AI-driven experiences across both on-device and server environments. This role also includes opportunities to open source your work. Join our team of highly skilled, impact-focused engineers!
Description
We're seeking strong senior machine learning research engineers to help build next-generation tools for large-scale deep learning. You'll join a team focused on accelerating training and inference speed, improving scalability, and advancing Apple's centralized ML platform. Candidates should bring polished coding skills and a passion for machine learning and computational science. We offer a respectful work environment, flexible responsibilities, and access to world-class experts and growth opportunities.
In this role, you will develop core components for our scalable ML platform, push the limits of existing training technologies, and create new techniques to overcome system constraints. Your work will be deployed on high-impact tasks across teams building Apple Intelligence products, with opportunities to open-source your contributions.
We are especially looking for PyTorch-focused ML experts driving system-level efficiency from on-device to large-scale models. If you have deep experience with PyTorch internals and high-performance ML infrastructure, we'd love to hear from you. We encourage releasing contributions as open source.
Minimum Qualifications
PhD or Master's degree in Computer Science, or equivalent industry experience, with 3+ years of experience in the AI/ML field.
Strong Python programming skills.
Solid understanding of software-hardware co-design principles and algorithms.
Solid understanding of the PyTorch software stack and experience maintaining state-of-the-art ML frameworks.
Strong understanding of LLM architectures and their core building blocks.
Preferred Qualifications
Experience working on AI/ML-optimized runtime stacks.
Familiarity with parallelization algorithms for large model training.
Familiarity with recent developments in foundation model architectures.
Experience with parallel training libraries such as PyTorch Distributed (torch.distributed), DeepSpeed, or FairScale.
Experience building ML models for on-device inference.
Publication record at ML conferences such as MLSys, NeurIPS, etc.

What Apple employees say

Pay

Benefits

Hours and flexibility

Workplace

Get the full story on Breakroom

Apple logo

About Apple

Sourced by ZipRecruiter

Imagine what you could do here! At Apple, new ideas have a way of becoming extraordinary products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish. Dynamic, intelligent people and inspiring, innovative technologies are the norm here. The people who work here have reinvented entire industries with all Apple Hardware products. The same real passion for innovation that goes into our products also applies to our practices strengthening our dedication to leave the world better than we found it.

Industry

Computer and electronic product manufacturing

Company size

10,000+ Employees

Headquarters location

Cupertino, CA, US

Year founded

1976

View All Apple Jobs

Apply