Company Overview

At TuriyamAI, we are pioneering world leading GenAI semiconductor solutions from India, for India and the World. Our breakthrough solutions are set to redefine the future of AI computing, driving unparalleled efficiency, performance, and accessibility for enterprises worldwide. :

We are looking for an exceptional engineers to develop and tune deep learning operations on highly parallel and custom instruction set of our next-generation AI hardware. The ideal candidate will have a strong background in heterogeneous computing environment, and experience with parallel programming, distributed algorithms to maximize hardware utilization at rack scale. Responsibilities:

Design, develop, optimize and validate math libraries for AI applications, focusing on dense linear algebra, matrix operations, and other relevant mathematical functions. These libraries should be highly efficient and scalable across variety of multi-modal AI models.

Work on optimizing the performance of AI math libraries to achieve hyper-optimized solutions.

Collaborate with hardware and software developers, to ensure seamless integration of math libraries with our SW stack

Stay updated with the latest advancements in AI and math library development. Conduct research to identify new techniques and technologies that can enhance our AI accelerators' performance and efficiency. Requirements:





Bachelor's, Master's or Ph.D. degree in Computer Science, Mathematics, Computer Engineering or a related field.

5+ years of proven experience in developing high-performance math libraries, preferably for AI applications.

Experience with GPU or AI accelerator programming

Experience writing efficient accelerator kernels for matmuls, convolutions, normalisation operators, kernel fusion, etc.

Understanding of HW architecture and must be comfortable learning new hardware architecture

Excellent problem-solving skills and ability to work in a blazing-fast-paced environment.

Preferred

Qualifications

Familiarity with machine learning frameworks such as PyTorch

Familiarity with compiler technology fundamentals such as: abstract syntax trees, control flows, data flow analysis, IR, and target lowering

Knowledge of assembly programming, low-level optimizations What We Offer:

· Competitive salary and benefits package.

· Prospect to work on cutting-edge AI technology.

· Collaborative and agile work environment.

· Professional growth and development opportunities.

How to Apply: Interested candidates are invited to submit their resume and a cover letter detailing their relevant experience and why they are a good fit for this role to [email protected]

📌 AI Accelerator Kernel Engineer (Bengaluru)
🏢 Turiyam AI
📍 Bengaluru