AI SW Stack Deployment Architect (12+ years)

Sandisk understands how people and businesses consume data and we relentlessly innovate to deliver solutions that enable today’s needs and tomorrow’s next big ideas. With a rich history of groundbreaking innovations in Flash and advanced memory technologies, our solutions have become the beating heart of the digital world we’re living in and that we have the power to shape.

Sandisk meets people and businesses at the intersection of their aspirations and the moment, enabling them to keep moving and pushing possibility forward. We do this through the balance of our powerhouse manufacturing capabilities and our industry-leading portfolio of products that are recognized globally for innovation, performance and quality.

Sandisk has two facilities recognized by the World Economic Forum as part of the Global Lighthouse Network for advanced 4IR innovations. These facilities were also recognized as Sustainability Lighthouses for breakthroughs in efficient operations. With our global reach, we ensure the global supply chain has access to the Flash memory it needs to keep our world moving forward.

Role Overview

We are looking for aSoftware Architect (12+ years experience) to lead theapplication/framework layer and deployment stack for the Next Generation Accelerator AI platform. This role owns how models run on Next Generation Accelerator—fromvLLM / PyTorch / TensotFlow/XLA to production deployment—ensuring correctness, performance, and scalability.

Key Responsibilities

• Architect integration ofvLLM, PyTorch, and TensorFlow, JAX/XLA into Next Generation Accelerator stack
• Defineframework → compiler → runtime APIs and contracts
• OwnLLM execution behavior (batching, KV cache, streaming inference)
• Design and implementend-to-end deployment workflows (packaging, versioning, reproducibility)
• Driveperformance optimization across model → framework → runtime
• Work cross-functionally withcompiler, runtime, and low-level SW teams
• Supportcustomer workloads, model onboarding, and debugging

Impact

Owncustomer-visible AI execution and deployment on Next Generation Accelerator, closing the gap betweenmodels and system performance, and enabling enterprise-grade AI solutions

Required Qualifications

• 10+ years inAI/ML systems or software architecture
• Strong experience withPyTorch / Transformers / LLMs
• Hands-on experience withLLM deployment and scalable inference engine systems e.g. vLLM, Triton, SGLang etc.
• Experience buildingscalable AI platforms (cloud/edge)
• Expertise insystem design, APIs, and cross-layer integration

Preferred Qualifications

• Experience withvLLM or similar LLM serving systems
• Familiarity withXLA / MLIR / compiler frameworks
• Exposure toAI accelerators (GPU/NPU) and runtime systems

Experience indistributed or multi-agent AI systems

Sandisk thrives on the power and potential of diversity. As a global company, we believe the most effective way to embrace the diversity of our customers and communities is to mirror it from within. We believe the fusion of various perspectives results in the best outcomes for our employees, our company, our customers, and the world around us. We are committed to an inclusive environment where every individual can thrive through a sense of belonging, respect and contribution.

Sandisk is committed to offering opportunities to applicants with disabilities and ensuring all candidates can successfully navigate our careers website and our hiring process. Please contact us at jobs.accommodations@sandisk.com to advise us of your accommodation request. In your email, please include a description of the specific accommodation you are requesting as well as the job title and requisition number of the position for which you are applying.