Job 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

Required 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
- Solid 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

Disclaimer: This job posting has been aggregated from external source. Role details, content, and availability are subject to change. Applicants are advised

📌 Sr. SW Test Engineer (Bengaluru)
🏢 Western Digital
📍 Bengaluru