新业务部-AI高性能算子开发工程师/专家

Job description

基于NPU平台，开发高性能计算库，支撑CV，NLP，大模型等模型运行，包括

• 完成高性能算子的需求分析与分解，满足NPU的各种应用场景；
• 与芯片/架构团队紧密配合，持续优化算子性能，逐步完善软硬件架构；
• 完成各类算子的软件方案设计及开发；
• 分析算子/网络的性能瓶颈，提出性能优化策略，达成性能目标；

Qualifications

• 图像处理、模式识别、机器学习等相关专业硕士及以上学历；
• 精通C++/Python软件编程语言；
• 熟悉Onnx/TensorFLow/Pytorch一种主流神经网络框架，熟悉各类算子；
• 熟悉并行/异构计算体系结构，在TPU/NPU/GPU一种平台有2年以上的开发和优化经验；
• 具备业界常用高性能库(TensorRT/cuDNN/CUDA/OPENCV)开发和性能优化经验者优先；
• 熟悉指令流水，计算机体系结构；
• 良好的团队沟通能力、技术攻关能力，责任心强。