Senior NPU AI Toolchain & Kernel Engineer

FLAMINGO RECRUITMENT PTE. LTD.

Singapore

Hybrid

SGD 120,000 - 180,000

Full time

14 days+

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Job summary

Flamingo Recruitment PTE. LTD. is seeking an experienced AI toolchain engineer to develop and optimize toolchains for NPU platforms.

You will implement and optimize AI operators and kernels, adapt models to hardware, and analyze bottlenecks to improve execution efficiency. Close collaboration with hardware, compiler, and algorithm teams is essential. The role requires a Master’s degree or above and 3+ years in related AI toolchain fields, with strong English and Chinese communication skills for

Qualifications

Master's degree or above in Computer Science, Electrical/Electronic Engineering, Automation, Mathematics, Physics, or related field.
3+ years of relevant experience in AI toolchain development, AI compiler engineering, AI operator/kernel development, or related fields.
Hands-on experience with NPU toolchains and understanding of AI algorithms and hardware-specific operator optimization.
Solid knowledge of AI model deployment and optimization on specialized AI hardware.
Experience in AI compiler development, operator/kernel optimization, or related research/projects is highly preferred.
Strong analytical, problem-solving, communication, and teamwork skills for fast-paced environments.
Bilingual proficiency in English and Chinese required to collaborate with China-based team.

Responsibilities

Develop, maintain, and optimize AI toolchains for chip-based NPU platforms.
Develop, implement, and optimize AI operators/kernels for efficiency and hardware performance.
Adapt AI models to NPU hardware and optimize performance and resource utilization.
Analyze bottlenecks and improve execution efficiency of AI workloads on NPU platforms.
Collaborate with hardware, compiler, and algorithm teams to deliver efficient AI computing solutions.