Staff Firmware Engineer

Location: Bengaluru, India

General Information
Job Name
Staff Firmware Engineer
Country/Region
India
City
Bengaluru
Work Location
Bangalore PTP Office (IBP)
Work Mode
Employees work on-site
Exemption Status
Exempt
Business Function
Firmware Engineering
Employment Type
Full-time
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
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 workloads execute on the platform
• Drive performance, scalability, and hardware utilization through optimized runtime and NN ops design

Required Qualifications

• 8+ years in systems software / runtime / performance engineering

Strong experience with:

• PyTorch / TensorFlow / JAX or similar frameworks
• NN operator/kernel development and optimization
• Operator fusion and graph-level optimizations

Hands-on expertise in:

• C/C++ and CUDA (or similar low-level programming)
• Runtime systems and execution engines

Strong understanding of:

• Memory hierarchy, data movement, and parallel execution

Preferred Qualifications:

• Experience with GPU/NPU or custom AI accelerators Familiarity with XLA / MLIR / compiler-runtime interaction
• Experience optimizing LLM or large-scale DL 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 at jobs.accommodations@sandisk.com 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.