Sr. SW Test Engineer - Neural Library (5-8 years)

Company Description

Sandiskunderstands 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.

Sandiskmeets people and businesses at the intersection of their aspirations and the moment, enabling them to keep moving and pushing possibilityforward. We do this through the balance of our powerhouse manufacturing capabilities and our industry-leading portfolio of products that are recognized globallyforinnovation, performance and quality.

Sandiskhas two facilities recognized by the World EconomicForum as part of the Global Lighthouse Networkforadvanced 4IR innovations. These facilities were also recognized as Sustainability Lighthousesforbreakthroughs 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 movingforward.

Job Description

We are looking for aSr. SW Test Engineer - Neural Library to validate neural network operators and libraries for anext-generation AI computational storage product. You will ensure correctness, performance, and robustness ofcomputekernels deployed on advanced hardware platforms.

Key Responsibilities

Validateneural network operators (matmul, convolution, attention, etc.)for correctness using reference modelsDevelop automated tests foroperator-level and graph-level validationPerformnumerical accuracy validationagainst frameworks likePyTorchor TensorFlowExecute and maintainsmoke and regression testsuitesBenchmark and validateperformance (throughput, latency, scaling behavior)Build and maintainperformance benchmarking frameworksPerformsoftware profiling to analyze execution bottlenecksConductmemory usage analysis and detect memory leaksIdentify issues acrosscompiler → runtime → kernel execution stackDebug failures acrosscompute kernels, runtime APIs, and hardware executionCollaborate with compiler, runtime, and hardware teams

Qualifications

Required Qualifications

4–6 yearsinAI/ML systems validation or performance testingStrong C++ programming skills (mandatory)Strong Object-Oriented Programming (OOP) fundamentals (mandatory)Experience withPython for test automationStrong understanding ofdeep learning operators and frameworks (PyTorchpreferred)Experience withnumerical validation and floating-point behaviorKnowledge ofparallel execution, memory hierarchy, and compute kernels

Preferred

Experience withprofiling and debugging tools (perf,VTune, Nsight, etc.)Experience withML execution stacks (MLIR/XLA or similar)Exposure toaccelerator validation (GPU/NPU/FPGA)Experience validatinglarge language model workloads

Additional Information

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 atjobs.accommodations@sandisk.comto 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.