🛠 Tech Stack

💼 About This Role

You'll join the team productizing the SW stack for our AI compute engine, developing software kernels for next-generation AI hardware. You'll map computational graphs from AI frameworks to the underlying architecture. This role offers the chance to work with compiler experts, ML, and hardware teams on cutting-edge AI.

🎯 What You'll Do

• Develop and enhance software kernels for AI hardware
• Map computational graphs from AI frameworks to hardware
• Optimize algorithms and trade-offs in HW-SW co-design
• Collaborate with compiler, ML, and hardware teams

📋 Requirements

• MS in CS/ECE/physics or related with 5+ years industry experience
• Strong grasp of computer architecture, data structures, system software
• Proficient in C/C++ and Python on Linux
• Experience implementing algorithms for FPGAs, DSPs, GPUs, or AI accelerators

✨ Nice to Have

• Prior startup or small team experience
• Experience with MLIR, LLVM, TVM, or Glow
• Experience with PyTorch or TensorFlow and ML models

🎁 Benefits & Perks

• 💰 Competitive salary ($190K–$300K)
• 📈 Equity and bonus included
• 🏢 Hybrid work (onsite 3-5 days/week in Santa Clara)
• 🤝 Inclusive culture and equal opportunity workplace
• 🏖️ Collaborative team environment