job posting · smartrecruiters
SANDISK CORP: AI Software Lead – PyTorch & CUDA Runtime (Next-Gen Accelerator) 10+ years
Observed on the employer’s careers system, last checked 2026-09-03.
- Employer
- SANDISK CORP
- Location
- Bengaluru
- Category
- Product Management
- Posted
- 2026-07-15
- Careers system
- smartrecruiters
- Observed
- 2026-09-03
Role description
Verbatim from the employer’s posting, rendered as plain text.
Role Overview We 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 Responsibilities Design and optimize NN operators for performance-critical workloads Develop new NN ops using CUDA/custom runtime APIs Drive runtime-level optimizations across compute, memory, and scheduling Own runtime ↔ NN layer interfaces and execution model Implement and optimize operator fusion (e.g., matmul + bias + LayerNorm) for efficient hardware utilization Identify and resolve performance bottlenecks across the stack Collaborate with compiler, PyTorch framework, and low-level SW teams Impact Own how efficiently AI…
Provenance
744000137940624