🛠 Tech Stack

💼 About This Role

You'll be part of the team productizing the SW stack for an AI compute engine and building software kernels for next-generation AI hardware. You'll map computational graphs from AI frameworks to underlying architecture and optimize hardware-software co-design.

🎯 What You'll Do

• Develop and maintain software kernels for AI hardware.
• Map computational graphs from AI frameworks to hardware architecture.
• Optimize hardware-software co-design and trade-offs.
• Collaborate with compiler, ML, and hardware teams.

📋 Requirements

• MS with 5+ years industry experience or PhD with 1+ year.
• Strong grasp of computer architecture, data structures, and ML fundamentals.
• Proficient in C/C++ and Python in Linux environments.
• Experience with CUDA or similar for specialized hardware (FPGAs, GPUs, AI accelerators).

✨ Nice to Have

• Prior startup or small team experience.
• Experience with MLIR, LLVM, TVM, or Glow.
• Experience with deep learning frameworks like PyTorch or TensorFlow.

🎁 Benefits & Perks

• 🤝 Collaborative culture with respect and direct communication.
• 🌍 Hybrid work with onsite in Bangalore 3-5 days/week.
• 🚀 Cutting-edge AI hardware and software innovation.