Intel AI Kernel Developer 

India, Karnataka, Bengaluru 

291111247

Job Description:

• We are looking for a dynamic and passionate contributors to work on Intel’s next generation GPUs.
• Works with internal engineering teams and stakeholders to deliver optimized complex math kernel operators for Intel GPU's.
• Partners with AI framework and workload engineers as needed to optimize endtoend AI models on Intel hardware features.

Role and responsibilities for this role includes:

• Optimizes performance of AI models through deep knowledge and expertise of AI frameworks, algorithms, models, and related hardware.
• Researches, develops, and adjusts new or existing complex math kernel code to address performance, numeric issues and modify operations to enhance performance with deterministic results.
• Serves as a trusted technical advisor and provides technical enabling.
• Partners with Intel software and hardware product development teams to accelerate and optimize future products in AI.

Qualifications:

• BTech, MS/MTech & PhD in CS, ECE or related fields with an overall experience of 6 to 12 years.
• Proficient in Advanced C++ (C++ 14/17) and Intermediate skills of Python and GPU programming.
• Experience in developing machine learning kernels such as GEMM, Flash attention, MoE, normalization etc.
• Strong understanding of probability theory, linear algebra and statistics.
• Well versed with numerical formats such as float32, float16, flopt8 etc and realted tradeoff's.
• Hands on experience in one of the frameworks such as PyTorch, Tensorflow or JAX.
• Experience in Matlabs Or Cuda.
• Practical knowledge working on deep learning models such as NLPs & LLMs
• Ability to debug complex issues in multi layered SW systems.
• Strong understanding of computer architecture and HW-SW optimization techniques.
• Effective communication skills and experience with working in a cross-geo teams.

• Preferable

  • Experience in developing, optimizing AI or related math kernels, and resolving related numeric issues.

  • Experience in developing, optimizing complex kernels operators like random number generators etc.

Experienced HireShift 1 (India)India, Bangalore

This role will require an on-site presence. * Job posting details (such as work model, location or time type) are subject to change.