Sandisk · 1 day ago
Performance Architect
Milpitas, CA
Full-time
Onsite
Mid, Senior Level
3+ years expSandisk is a leading innovator in Flash and advanced memory technologies, dedicated to delivering solutions that meet today's data needs. The Performance Architect will develop advanced system architectures and complex simulation models for next-generation products, focusing on enhancing performance and efficiency in high-bandwidth flash technologies.
Data StorageManufacturingSemiconductor
Responsibilities
Build SystemC performance models for HBF based products covering end-to-end from GPU/TPU/NPU/xPU, host interface, memory hierarchy, basedie controller, and HBF using various packaging technolgies
Responsible for improving the AI/ML ASIC Architecture performance through hardware & software co-optimization, post-silicon performance analysis, and influencing the strategic product roadmap
Workload analysis and characterization of ASIC and competitive datacenter and AI solutions to identify opportunities for performance improvement in our products
Collaboration with Architecture team to resolve performance issues and optimize the performance and TCO of their HBF based datacenter technologies
Experience modeling one or some components of AI/ML accelerator ASICs such as HBM, PCIe/UCIe/CXL, NoC, DMA, Firmware Interactions, NAND, xPU, fabrics, etc
Performance modeling and optimization for multi-trillion parameter LLM training/inference including Dense, Mixture of Experts (MoE) with multiple modalities (text, vision, speech)
Model/optimize novel parallelization strategies across tensor, pipeline, context, expert and data parallel dimensions
Architect memory-efficient training systems utilizing techniques like structured pruning, quantization (MX formats), continuous batching/chunked prefill, speculative decoding
Incorporate and extend SOTA models such as GPT-4, Reasoning models like Deepseek-R1, and multi-modal architectures
Collaborate with internal and external stakeholders/ML researchers to disseminate results and iterate at rapid pace
Qualification
Required
Bachelors or Masters or PhD in Computer/Electrical Engineering with 3+ years of relevant experience in Performance Modeling, Simulation, and Analysis using SystemC
Preferred
At least 3+ years of experience with SystemC modeling
Deep experience optimizing large-scale ML systems, GPU architectures
Good understanding of computer/graphics architecture, ML, LLM
Strong track record of technical leadership in GPU performance and workload analysis
Expert knowledge of transformer architectures, attention mechanisms, and model parallelism techniques
Experience with GPU or TPU and system microarchitecture
Proficiency in principles and methods of microarchitecture, software, and hardware relevant to performance engineering
Capable of developing wide system view for complex AI/ML Accelerator ASIC systems
Proficiency with SoC and system performance analysis fundamentals, tools, and techniques including hardware performance monitors and PERF profiling
Familiar with IO subsystem microarchitecture performance modeling and background in NVMe/PCIe//UCIe/CXL/NVLink microarchitecture and protocols is a plus
Experience of simulation using System C and TLM, behavioral modeling and performance analysis - advantage
Multi-disciplinary experience, including familiarity with Firmware and ASIC design
PyTorch, CUDA, TensorRT, OpenAI Triton, and ONNX
Distributed systems: Ray, Megatron-LM
Performance analysis tools: NSight Compute, nvprof, PyTorch Profiler
KV cache optimization, Flash Attention, Mixture of Experts
High-speed networking: InfiniBand, RDMA, NVLink
Expertise in CUDA programming, GPU memory hierarchies, and hardware-specific optimizations
Proven track record architecting distributed training systems handling large scale systems
Previous experience with storage systems, protocols, and NAND flash – advantage
Experience with datacenter and AI workload analysis and optimization
Experience with multi-core systems and multi-thread interactions
Experience analyzing and optimizing workloads
Benefits
Paid vacation time
Paid sick leave
Medical/dental/vision insurance
Life, accident and disability insurance
Tax-advantaged flexible spending and health savings accounts
Employee assistance program
Other voluntary benefit programs such as supplemental life and AD&D, legal plan, pet insurance, critical illness, accident and hospital indemnity
Tuition reimbursement
Transit
The Applause Program
Employee stock purchase plan
Sandisk's Savings 401(k) Plan
Company
Sandisk
SanDisk designs, develops, manufactures, and markets data storage solutions in the United States and internationally.
5001-10000 employees
H1B Sponsorship
Sandisk has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (210)
Funding
Current Stage
Public CompanyTotal Funding
$713.58M2025-06-06Post Ipo Secondary· $713.58M
2015-10-21Acquired
1995-11-17IPO
Leadership Team
N
Netanel Nutman
Senior Manager , Software/Protocol Testing
Recent News
2026-01-24
Investing.com
2026-01-24
Company data provided by crunchbase