ZipRecruiter

Log In

1

Senior Pytorch Jobs in California (NOW HIRING)

The Marlin Alliance

Sr Machine Learning Engineer

San Diego, CA · On-site

$131K - $173K/yr

The Senior ML Engineer will collaborate with multidisciplinary teams to deliver mission-focused AI ... Implement ML solutions using frameworks such as TensorFlow, PyTorch, and scikit-learn. * Contribute ...

Teleo

Senior Perception Engineer

Palo Alto, CA · On-site

$123K - $168K/yr

The Senior Perception Engineer will build perception algorithms and architecture to ensure safe ... PyTorch • Experience shipping perception or ML systems into production • Systems thinker ...

Workday, Inc.

Senior/Principal AI Engineer

Pleasanton, CA · On-site

$139K - $192K/yr

... Pytorch, TensorFlow * 6+ years of professional experience in building services to host machine ... Senior AI Engineer Basic Qualifications * 7+ years experience as a member of a data science ...

Workday

Senior/Principal AI Engineer

Pleasanton, CA · On-site

$139K - $192K/yr

... Pytorch, TensorFlow * 6+ years of professional experience in building services to host machine ... Senior AI Engineer Basic Qualifications * 7+ years experience as a member of a data science ...

Alterion

Senior Applied AI Engineer

San Francisco, CA · On-site

$123K - $169K/yr

... PyTorch or similar. Preferred : • Experience with GenAI evaluation, observability, or safety tooling. • Familiarity with embeddings, vector search, or semantic similarity systems. • Exposure to ...

Qualcomm

Sr. AI Engineer

Santa Clara, CA · On-site

$122K - $168K/yr

Preferred : • Bachelor's or Master's degree in Computer Science, Machine Learning, Data Science, or related field. • Solid experience training machine learning models with frameworks like PyTorch ...

Apple

Senior Applied Researcher

San Francisco, CA · On-site

$107K - $147K/yr

Description We are seeking a Senior Applied Researcher with demonstrated expertise in Generative AI ... Proficiency in Python and ML frameworks such as PyTorch or TensorFlow, with experience deploying ...

Imagry

Senior Deep Learning Engineer - Perception

San Jose, CA · On-site

$123K - $169K/yr

Senior Deep Learning Engineer, Computer Vision imagry.E4.E30@comeetapply.com Location: San Jose, CA ... Hands-on experience with deep learning frameworks such as PyTorch * Strong mathematical aptitude ...

previous page

Showing results 41-60

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 HPC & GPU Infrastructure Engineer

Sciforium

San Francisco, CA • On-site

$150K - $220K/yr

Full-time

Medical, Dental, Vision, Retirement

Re-posted 4 days ago

Job description

Sciforium is an AI infrastructure company developing next-generation multimodal AI models and a proprietary, high-efficiency serving platform. Backed by multi-million-dollar funding and direct sponsorship from AMD with hands-on support from AMD engineers the team is scaling rapidly to build the full stack powering frontier AI models and real-time applications.
About the role
We are seeking a Senior HPC & GPU Infrastructure Engineer to take full ownership of the health, reliability, and performance of our GPU compute cluster. You will be the primary custodian of our high-density accelerator environment and the linchpin between hardware operations, distributed systems, and machine learning workflows. This role spans everything from hands-on Linux systems engineering and GPU driver bring-up to maintaining the ML software stack (CUDA/ROCm, PyTorch, JAX, vLLM). If you love squeezing every bit of performance out of hardware, enjoy debugging GPUs at scale, and want to build world-class AI infrastructure, this role is for you.
What you'll do
1. System Health & Reliability (SRE)
  • On-Call Response: Act as the primary responder for system outages, GPU failures, node crashes, and cluster-wide incidents. Minimize downtime by resolving issues rapidly.
  • Cluster Monitoring: Implement and maintain monitoring for GPU health, thermal behavior, PCIe/NVLink topology issues, memory errors, and overall system load.
  • Vendor Liaison: Coordinate with data center staff, hardware vendors, and on-site technicians for repairs, RMA processing, and physical maintenance of the cluster.

2. Linux & Network Administration
  • OS Management: Install, patch, and maintain Linux distributions (Ubuntu / CentOS / RHEL). Ensure consistent configuration, kernel tuning, and automation for large node fleets.
  • Security & Access Controls: Configure VPNs, iptables/firewalls, SSH hardening, and network routing to secure our computer infrastructure.
  • Identity & Storage Management: Manage LDAP/FreeIPA/AD for user identity, and administer distributed file systems such as NFS, GPFS, or Lustre.

3. GPU & ML Stack Engineering
  • Deployment & Bring-Up: Lead deployment of new GPU nodes, including BIOS configuration, NUMA tuning, GPU topology validation, and cluster integration.
  • Driver & Kernel Management: Build and optimize kernel modules, maintain GPU drivers and runtime stacks for both NVIDIA (CUDA) and AMD (ROCm).
  • Software Stack Maintenance: Maintain and optimize ML frameworks and libraries PyTorch, JAX, CUDA toolkit, cuDNN, ROCm, NCCL, and supporting runtime systems.
  • Advanced Debugging: Troubleshoot complex interactions involving GPUs, compilers, ML frameworks, and distributed training runtimes (e.g., vLLM compilation failures, CUDA memory leaks, ROCm kernel crashes).

Ideal candidate profile
  • 5+ years of experience in HPC, GPU cluster operations, Linux systems engineering, or similar roles.
  • Bachelor's or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or a related technical field.
  • Strong expertise with NVIDIA (H100/B200) or AMD (MI325x/MI355x) GPUs, including driver and kernel-level debugging.
  • Deep understanding of Linux internals, kernel modules, hardware bring-up, and systems performance tuning.
  • Experience with network security, including VPNs, iptables/firewalld, SSH, and identity management (LDAP/FreeIPA/AD).
  • Proficiency in Bash and Python for scripting, automation, and workflow tooling.
  • Familiarity with ML software stacks: CUDA toolkit, cuDNN, NCCL, ROCm, JAX/PyTorch runtime behavior.
  • Deep debugging experience with NVLink/NVSwitch fabrics and RDMA networking.

Nice-to-have
  • Experience with job schedulers such as Slurm, Kubernetes, or Run:AI.
  • Exposure to vLLM, model serving optimizations, or inference systems.
  • Hands-on experience with configuration management tools (Ansible, SaltStack, Terraform).
  • Previous experience supporting ML research teams in a startup or research-heavy environment.

Benefits include
  • Medical, dental, and vision insurance
  • 401k plan
  • Daily lunch, snacks, and beverages
  • Flexible time off
  • Competitive salary and equity

Equal opportunity
Sciforium is an equal opportunity employer. All applicants will be considered for employment without attention to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.

Apply