AI Toolchain Algorithm Engineer (NPU)

FLAMINGO RECRUITMENT PTE. LTD.; Singapore; Hybrid; SGD 120,000 - 180,000; Full time

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.

Skills

AI toolchain dev

AI compiler engineering

Operator/kernel development

AI model deployment

Hardware optimization

Debugging & optimization

Education

Master's degree or above in CS/EE/Automation/Math/Physics

Tools

NPU toolchains

AI compiler tooling

Hardware-specific operator optimization

Job description

Location: Singapore, flexible location (Work from Home or Office)

Key Responsibilities:

1. Develop, maintain, and optimize AI toolchains for chip-based NPU platforms.

2. Develop, implement, and optimize AI operators/kernels to improve computational efficiency and hardware performance.

3. Adapt AI models to NPU hardware and optimize model performance, computational efficiency, and resource utilization.

4. Analyze computational bottlenecks and improve the execution efficiency of AI workloads on NPU platforms.

5. Work closely with hardware, compiler, and algorithm teams to deliver efficient AI computing solutions.

Requirements:

1. Master's degree or above in Computer Science, Electrical/Electronic Engineering, Automation, Mathematics, Physics, or a related discipline.

2. Has 3+ years of relevant experience in AI toolchain development, AI compiler engineering, AI operator/kernel development, or related fields.

3. Hands-on experience with NPU toolchains and a strong understanding of AI algorithms and hardware-specific operator optimization.

4. Solid knowledge of AI model deployment and optimization on specialized AI hardware.

5. Experience in AI compiler development, operator/kernel optimization, or related research and projects is highly preferred.

6. Strong analytical, problem-solving, communication, and teamwork skills, with the ability to work effectively in a challenging and fast-paced environment.

7. Bilingual proficiency in English and Chinese is required to communicate and collaborate effectively with the China-based team.