AI/ML Compiler Developer (NPU Acceleration)

About the role

As a C++ Kernel Developer, you will play a crucial role in designing, optimizing, and implementing machine learning kernels specifically tailored for vector processors. Your work will directly impact the efficiency, speed, and accuracy of our machine learning models. Collaborate with the research and software teams to integrate these kernels into the existing software stack.

What you'll do

• Play a crucial role in designing, optimizing, and implementing machine learning kernels specifically tailored for vector processors
• Collaborate with the research and software teams to integrate these kernels into the existing software stack
• Develop vectorized code that leverages SIMD (Single Instruction, Multiple Data) and ILP (instruction level parallelism) for maximum performance
• Identify bottlenecks and optimize critical sections for better throughput
• Testing and Validation: Develop CPU models for the ML operators in C++/ Python to validate accuracy

What you need

• 5–9 years in data science & ai
• Machine learning
• Python
• LLMs
• C++
• C
• Git
• PhD or B.Tech or B.E. or B.Sc or BCA or M.Tech or MCA

Good to have

• TensorFlow
• PyTorch