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

Sandisk; Bengaluru; On-site; INR 1,200,000 - 1,800,000; Full time

Job summary

Sandisk in Bengaluru is seeking a Sr. SW Test Engineer - Neural Library to validate neural network operators and libraries for a next-generation AI storage product. You will ensure correctness, performance and robustness of compute kernels on advanced hardware.

The role focuses on operator-level and graph-level validation, numerical accuracy, and automated test development, with collaboration across compiler, runtime and hardware teams to optimize throughput and memory usage.

Qualifications

• 4-6 years in AI/ML systems validation or performance testing.
• Strong C++ programming skills (mandatory).
• Strong OOP fundamentals (mandatory).
• Experience with Python for test automation.
• Strong understanding of deep learning operators and frameworks (PyTorch preferred).
• Experience with numerical validation and floating-point behavior.
• Knowledge of parallel execution, memory hierarchy, and compute kernels.

Responsibilities

• Validate neural network operators (matmul, convolution, attention, etc.) for correctness using reference models.
• Develop automated tests for operator-level and graph-level validation.
• Perform numerical accuracy validation against frameworks like PyTorch or TensorFlow.
• Execute and maintain smoke and regression test suites.
• Benchmark and validate performance (throughput, latency, scaling behavior).
• Build and maintain performance benchmarking frameworks.
• Perform software profiling to analyze execution bottlenecks.
• Conduct memory usage analysis and detect memory leaks.
• Identify issues across compiler → runtime → kernel execution stack.
• Debug failures across compute kernels, runtime APIs, and hardware execution.
• Collaborate with compiler, runtime, and hardware teams.

Skills

C++ programming

OOP fundamentals

Python for test automation

Deep learning operators understanding

Numerical validation

Parallel execution & memory hierarchy

Tools

Perf tools

VTune

Nsight

Job description

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.

Description

We are looking for a Sr. SW Test Engineer - Neural Library to validate neural network operators and libraries for a next-generation AI computational storage product. You will ensure correctness, performance, and robustness of compute kernels deployed on advanced hardware platforms.

Key Responsibilities

• Validate neural network operators (matmul, convolution, attention, etc.) for correctness using reference models
• Develop automated tests for operator-level and graph-level validation
• Perform numerical accuracy validation against frameworks like PyTorch or TensorFlow
• Execute and maintain smoke and regression test suites
• Benchmark and validate performance (throughput, latency, scaling behavior)
• Build and maintain performance benchmarking frameworks
• Perform software profiling to analyze execution bottlenecks
• Conduct memory usage analysis and detect memory leaks
• Identify issues across compiler → runtime → kernel execution stack
• Debug failures across compute kernels, runtime APIs, and hardware execution
• Collaborate with compiler, runtime, and hardware teams

Qualifications

• 4-6 years in AI/ML systems validation or performance testing
• Strong C++ programming skills (mandatory)
• Strong Object-Oriented Programming (OOP) fundamentals (mandatory)
• Experience with Python for test automation
• Strong understanding of deep learning operators and frameworks (PyTorch preferred)
• Experience with numerical validation and floating-point behavior
• Knowledge of parallel execution, memory hierarchy, and compute kernels

Preferred

• Experience with profiling and debugging tools (perf, VTune, Nsight, etc.)
• Experience with ML execution stacks (MLIR/XLA or similar)
• Exposure to accelerator validation (GPU/NPU/FPGA)
• Experience validating large language model workloads

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 [e-mail] 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.