Sandisk logoSandisk logo

AI/ML ASIC Architect at Sandisk | Milpitas

SandiskMilpitas
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Experience Level

Mid to Senior

Qualifications

Required Qualifications:Bachelor's, Master's, or PhD in Computer or Electrical Engineering with a minimum of 15 years of hands-on architecture experience in authoring specifications. Strong background in architecting ASIC, SoC, or I/O subsystems with a focus on PCIe/UCIe/CXL and DMA engines. In-depth knowledge of I/O subsystem and DMA interactions with internal embedded processor subsystems (x86, RISC-V, or ARM) and external host CPUs. Solid understanding of computer and graphics architecture, machine learning, and large language models (LLM). Experience in architecting GPU/TPU/xPU Accelerator systems, specifically with high bandwidth memory hierarchies optimized for multi-trillion parameter LLM training/inference, including Dense and Mixture of Experts (MoE) with various modalities (text, vision, speech). Expertise in KV cache optimization, Flash Attention, and Mixture of Experts. Proven experience in optimizing large-scale ML systems and GPU architectures. Strong proficiency in microarchitecture principles and methods.

About the job

Join Sandisk as an AI/ML ASIC Architect where you will spearhead the development of cutting-edge AI Storage Solutions. This role involves designing advanced system architectures and specifications for the next generation of AI/ML Accelerator ASIC products. You will play a pivotal role in driving and analyzing the frontend architecture of these innovative products.

Your expertise will guide new architectural initiatives that incorporate state-of-the-art frontend interfaces such as UCIe, PCIe, and CXL, integrating AI Storage Solutions with xPU in a sophisticated 3D packaging system. Your collaboration with design and product planning teams will focus on delivering innovative and competitive adaptive accelerator solutions.

Key responsibilities include authoring comprehensive architecture specifications, collaborating with fellow architects, and working alongside RTL/DV/Simulation/Emulation/Firmware teams to evaluate performance, power, area, and endurance metrics. You will tackle complex challenges and develop transformative products that redefine the data-centric architecture paradigm.

About Sandisk

At Sandisk, we are dedicated to understanding how individuals and businesses leverage data, driving relentless innovation to meet today’s needs and tomorrow’s aspirations. With a rich legacy of groundbreaking advancements in Flash and advanced memory technologies, our solutions are integral to the digital landscape, empowering the future of innovation. Sandisk operates at the nexus of customer ambitions and real-time solutions, enabling progress through a blend of robust manufacturing capabilities and a globally recognized portfolio of products celebrated for their innovation, performance, and quality. Our facilities are acknowledged by the World Economic Forum as part of the Global Lighthouse Network, recognized for advanced 4IR innovations, and have also earned accolades as Sustainability Lighthouses for operational efficiencies.

Similar jobs

Browse all companies, explore by city & role, or SEO search pages. View directory listings: all jobs, search results, location & role pages.

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.