AI Engineer – NPU Modeling & Optimization

Position Overview:

Black Sesame Technologies is seeking motivated and talented individuals to join our team as Engineer, AI Computing . This position offers an exciting opportunity to work on cutting-edge technologies for autonomous driving. You’ll contribute to the development of high-performance simulation models and optimization strategies for NPU HW architecture and helping us to deploy AI solutions on embedded platforms. We welcome individuals who are eager to learn, innovate, and make a real-world impact.

Responsibilities:

• Develop performance analysis models to identify hardware bottlenecks and propose architectural improvements.
• Build functional and execution models to accurately simulate chip behaviours and results.
• Collaborate with design (DE) and verification (DV) teams to build functional and performance models of chip modules, and validate model accuracy against RTL.
• Work with the compiler/model compilation team to provide optimization feedback and ensure accurate simulation results.
• Develop custom CUDA operators in PyTorch that maintain data consistency between model and hardware, for customer-facing applications.

Qualification/ Requirements:

• Education: Master’s or PhD degree in Computer Science, Electrical/Computer Engineering, or related fields.
• Experience: Fresh graduates are welcome! Internship, academic research, or hands-on project experience in chip modeling, performance simulation, or AI computing will be a strong advantage.
• Programming Skills: Proficiency in Python, CUDA, and deep learning frameworks such as PyTorch. Experience with C++ or hardware modelling tools is a plus.
• Problem-Solving: Strong interest in system architecture and performance analysis. Able to identify and resolve modelling discrepancies.
• Teamwork: Effective communicator and team player, capable of working across multidisciplinary teams.