AI/ML ASIC Architect

Responsibilities

• Drive AI/ML ASIC architecture integrating AI Storage with GPU/TPU/xPU accelerators via UCIe, PCIe, or CXL
• Author architecture specifications for AI/ML xPU based accelerators
• Define I/O subsystem and PCIe DMA architectures, including interactions with embedded processors and Network on Chip
• Create modular I/O subsystem architectures for Chiplet, monolithic, or 3D form factors
• Collaborate with cross-functional teams to scope SoC requirements and analyze PPA tradeoffs
• Perform LLM workload analysis and characterization to identify performance improvement opportunities
• Architect memory-efficient inference/training systems using pruning, quantization, and speculative decoding

Requirements

• Bachelors, Masters, or PhD in Computer or Electrical Engineering
• 15+ years of hands-on Architecture experience authoring specifications
• Strong background in ASIC, SoC, or I/O subsystems (PCIe/UCIe/CXL and DMA engines)
• Deep understanding of computer architecture, ML, and LLM
• Experience architecting GPU/TPU/xPU accelerator systems with high bandwidth memory hierarchy
• Proficiency in microarchitecture, software, and hardware performance engineering
• Knowledge of ARM Processors and AXI Interconnects

Preferred Qualifications

• Familiarity with UCIe, CXL, NVLink, or UAL protocols
• Experience with high-speed networking like InfiniBand or RDMA
• Expert knowledge of transformer architectures and attention mechanisms
• Expertise in CUDA programming and GPU memory hierarchies
• Experience with NVMe storage systems, protocols, and NAND flash

About the Company

Sandisk delivers innovative Flash and advanced memory technologies that serve as the beating heart of the digital world, enabling people and businesses to push the boundaries of possibility.