Staff AI/HPC Infrastructure Architect

SanDisk

Milpitas (CA)

On-site

USD 154,000 - 261,000

Full time

2 days ago
Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Sandisk is seeking a Staff Engineer AI/HPC Infrastructure in Milpitas, CA to design and operate GPU-based AI platforms for training, inference, and research. The role focuses on AI infrastructure, LLM serving, memory architecture, and storage systems, shaping next-generation AI memory products.

You will build AI testbeds, deploy serving platforms, and characterize workloads while collaborating with academia and industry partners. On-site, full-time role with competitive compensation.

Qualifications

  • BS+ 10 years or MS+ 6 years or PhD+ 3 years in relevant AI experience.
  • Relevant AI experience required for BS/MS/PhD track.

Responsibilities

  • Build and Modify AI/HPC Research Infrastructure: 8-100+ GPU environments and supporting server/network/storage/software.
  • Deploy and Optimize LLM Serving Platforms: install and optimize vLLM, NVIDIA Dynamo, TensorRT-LLM, Triton, SGLang.
  • AI Workload Characterization: capture workload traces, analyze memory bandwidth, KV-cache growth and storage access.
  • AI Memory and Storage Architecture Evaluation: evaluate bottlenecks across SSDs, networking, and GPU interconnects.
  • Digital Twin and Simulation: build sim environments and digital twins; model future AI infra configurations.
  • Research and Ecosystem Collaboration: collaborate with universities, startups, labs, cloud providers, OEMs.

Skills

Linux
Kubernetes
Docker
Slurm
Python
GPU cluster administration
Distributed systems engineering

Education

BS+10 years or MS+6 years or PhD+3 years in relevant field

Tools

Linux
Kubernetes
Docker
Slurm
Python

Job description

Sandisk is seeking a Staff Engineer AI/HPC Infrastructure in Milpitas, CA to design and operate GPU-based AI platforms for training, inference, and research. The role focuses on AI infrastructure, LLM serving, memory architecture, and storage systems, shaping next-generation AI memory products.

You will build AI testbeds, deploy serving platforms, and characterize workloads while collaborating with academia and industry partners. On-site, full-time role with competitive compensation.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Senior AI/HPC Systems Architect
Senior AI/HPC Systems Architect

SanDisk • Milpitas (CA)

On-site
USD 180,000 - 306,000
Senior AI Systems Architect — Hybrid HW/SW Co‑Design
Senior AI Systems Architect — Hybrid HW/SW Co‑Design

SanDisk • Milpitas (CA)

Hybrid
USD 180,000 - 306,000
Principal AI Systems & Performance Lab Architect
Principal AI Systems & Performance Lab Architect

SanDisk • Milpitas (CA)

On-site
USD 202,000 - 344,000
Staff Engineer AI/HPC Infrastructure
Staff Engineer AI/HPC Infrastructure

SanDisk • Milpitas (CA)

On-site
USD 154,000 - 261,000
Staff Engineer, AI Compute & HPC Infrastructure
Staff Engineer, AI Compute & HPC Infrastructure

ATOMS Careers page • San Francisco (CA)

On-site
USD 224,000 - 284,000
Medical, Dental, Vision
401(k)
Equity
+2
Staff Engineer — AI & HPC Compute Infra
Staff Engineer — AI & HPC Compute Infra

Atoms • San Francisco (CA)

On-site
USD 224,000 - 284,000
Medical, Dental, Vision
Equity
401(k)
+3
Senior staff AI/HPC Infrastructure Engineer
Senior staff AI/HPC Infrastructure Engineer

SanDisk • Milpitas (CA)

On-site
USD 180,000 - 306,000
Hardware Systems Engineer, AI & HPC NPI
Hardware Systems Engineer, AI & HPC NPI

Meta • Menlo Park (CA)

Hybrid
USD 140,000 - 210,000
Staff Engineer, GPU HPC & AI Compute Infra
Staff Engineer, GPU HPC & AI Compute Infra

Pronto • San Francisco (CA)

On-site
USD 224,000 - 284,000
Medical, Dental, Vision
Equity
Sick Time and Paid Holidays
+1
Senior HPC & AI Infrastructure Architect
Senior HPC & AI Infrastructure Architect

Advanced Micro Devices, Inc. • San Jose (CA)

On-site
USD 180,000 - 240,000
Benefits at a glance