Sr Software Engineer – Kernel

<p><strong>Location:&nbsp;Hybrid, working onsite at our Bengaluru, India offices 3-5 days per week.<br></strong></p> <p><strong>What you will do:</strong></p> <p><strong>&nbsp;</strong>The role requires you to be part of the team that helps productize the SW stack for our AI&nbsp;compute engine. As part of the software team, you will be responsible for the development,</p> <p>enhancement, and maintenance of software kernels for next-generation AI hardware. You&nbsp;possess experience building software kernels for HW architectures. You possess a very strong</p> <p>understanding of various hardware architectures and how to map algorithms to the architecture.&nbsp;You understand how to map computational graphs generated by AI frameworks to the</p> <p>underlying architecture. You have had past experience working across all aspects of the full&nbsp;stack toolchain and understand the nuances of what it takes to optimize and trade off various</p> <p>aspects of hardware-software co-design. You can build and scale software deliverables in a&nbsp;tight development window. You will work with a team of compiler experts to build out the</p> <p>compiler infrastructure, working closely with other software (ML, systems) and hardware (mixed&nbsp;signal, DSP, CPU) experts in the company.</p> <p><strong>What you will bring:</strong><strong>&nbsp;</strong></p> <p><strong>Minimum</strong>:</p> <p>MS in computer engineering, math, physics, or a related degree with 5+ years of industry&nbsp;experience or PhD in computer engineering, math, physics, or a related degree with 1+ years&nbsp;of industry experience</p> <p>Strong grasp of computer architecture, data structures, system software, and machine&nbsp;learning fundamentals.<br>Proficient in C/C++ and Python development in Linux environments and using standard&nbsp;development tools.<br>Experience implementing algorithms in high-level languages such as C/C++ and Python.<br>Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs,&nbsp;and AI accelerators using libraries such as CUDA, etc.&nbsp;<br>Experience in implementing operators commonly used in ML workloads—GEMMs, Convolutions, BLAS, SIMD operators for operations like softmax, layer normalization, pooling, etc.</p> <p>Experience with development for embedded SIMD vector processors such as Tensilica.<br>Self-motivated team player with a strong sense of ownership and leadership.</p> <p><strong>Preferred:</strong></p> <p>Prior startup, small team, or incubation experience.</p> <p>Experience with ML frameworks such as TensorFlow and/orPyTorch.</p> <p>Experience working with ML compilers and algorithms, such as MLIR, LLVM, TVM, Glow, etc.</p> <p>Experience with a deep learning framework (such as PyTorch or TensorFlow) and ML models&nbsp;for CV, NLP, or recommendation</p> <p>Work experience at a cloud provider or AI compute/subsystem company.<br><br><br></p>