← Back to opportunities

AI Kernel Engineer

📍 Location
Pune
⏰ Job Type
fullTime
📅 Posted
June 06, 2026

About the Role

You will play a key role in building and optimising high-performance AI kernels for a next-generation compute platform. This role focuses on enabling efficient execution of AI and LLM workloads by developing, profiling, and optimising kernels across varying hardware configurations.

You will be responsible for:

  • Developing AI/LLM kernels and operators for efficient inference on a specialised compute platform
  • Optimising kernel performance across different hardware configurations and workloads
  • Profiling and analysing performance across compute, memory, and parallelism to identify bottlenecks
  • Optimising low-level C/C++ code to maximise hardware utilisation
  • Collaborating across the AI inference stack, including runtime, compiler, and system layers
  • Contributing to improvements in toolchain, compiler, and runtime components
  • Supporting internal teams and external stakeholders with technical insights and documen...

Ready to Join Through a Referral?

Apply now and get connected directly with the hiring team

Apply for this Position