Principal AI/HPC Software Engineer

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.

The goal is to demonstrate and optimize future memory-centric architectures that leverage NAND, storage systems, and emerging AI memory technologies. Core ResponsibilitiesAI Infrastructure Software Development Develop System Software And Prototype Platforms For AI/ML InfrastructureAI inference platformsAI training environments Memory-aware computing platforms Distributed AI systems Open-source AI infrastructure The role specifically calls out using computer systems and open-source software technologies to demonstrate next-generation memory architectures. Memory-Centric Software Architecture Design Software Supporting Memory management Cache hierarchy optimization Data movement protocols Storage-memory integrationAI system runtime environments The engineer develops software that exposes the advantages of future memory technologies within AI systems. Hardware/Software Co-DesignWork With Architecture Teams ToEvaluate new memory technologies Prototype future AI systems Optimize performance and power efficiency Validate next-generation AI memory concepts The role explicitly requires close collaboration with hardware architects. Frameworks and APIsDevelopInfrastructure software frameworks Runtime environmentsAPIsSystem software libraries Benchmarking and validation toolsfor future AI systems. Required ProgrammingPythonCC++Systems Operating systems Device drivers System software design System programmingAI ComputingGPU architectureCUDA programmingAI infrastructure Computer architecture Memory systemsAI Lab Software Engineer The Strongest Candidate Would HaveAI ServingvLLM; TensorRT-LLM; Triton; SGLang; Dynamo Distributed AIRay; Kubernetes; Docker; SlurmGPU SystemsCUDA; NCCLMulti-GPU programming High-Speed InfrastructureRDMA; RoCEInfiniBandMemory-System ResearchHBFKV-cache hierarchy Storage-memory integrationAI SSD architectures Research Capabilities Workload tracing System simulation Digital twinsAI infrastructure modeling The AI Lab focused on HBF, KV-cache offload, AI SSDs, and memory-system innovation, this software engineer would essentially function as a LLM Infrastructure + AI Systems Research Engineer, rather than a traditional application software developer.