Sr. SW Test Engineer

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

• 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

  Disclaimer: This job posting has been aggregated from external source. Role details, content, and availability are subject to change. Applicants are advised to confirm the latest information directly on the company website before applying.