Company DescriptionSandisk 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.Job DescriptionRole OverviewWe are looking for a Software Lead (8+ years’ experience) to own the runtime and neural network (NN) layer of a next-generation AI accelerator platform. This role focuses on designing, optimizing, and implementing NN operators and developing new ops using CUDA/custom runtime APIs to deliver high-performance execution on custom AI hardware.Key ResponsibilitiesDesign and optimize NN operators for performance-critical workloadsDevelop new NN ops using CUDA/custom runtime APIsDrive runtime-level optimizations across compute, memory, and schedulingOwn runtime ↔ NN layer interfaces and execution modelImplement and optimize operator fusion (e.g., matmul + bias + LayerNorm) for effective hardware utilizationIdentify and resolve performance bottlenecks across the stackCollaborate with compiler, PyTorch framework, and low-level SW teamsImpactOwn how

📌 AI Software Lead – PyTorch (Bengaluru)
🏢 Sandisk
📍 Bengaluru