岗位描述
岗位职责:
1. 负责AI模型在昇腾平台上的适配、开发和性能优化,打通模型转换、编译、量化、部署及推理链路。
2. 负责PyTorch、MindSpore、ONNX等模型迁移,解决算子不支持、精度偏差和性能瓶颈等问题。
3. 基于CANN、AscendCL、ATC、Ascend C/TBE、MindIE等开发模型适配工具和自定义算子。
4. 负责双卡及多卡推理优化,支持大模型、视觉模型等在边缘AI服务器上的部署。
5. 参与鲲鹏+昇腾AI工具链建设及行业生态适配,支持产品POC和客户项目交付。
任职要求:
1. 本科及以上学历,计算机、人工智能、电子信息等相关专业,5年以上相关经验。
2. 熟练掌握Python、C/C++,熟悉Linux及ARM64开发环境。
3. 熟悉主流AI框架及模型推理流程,具备模型迁移、量化、算子开发或性能调优经验。
4. 熟悉昇腾AI软件栈,有CANN、AscendCL、Ascend C/TBE或MindIE实战经验。
5. 具备较强的问题定位、方案设计和跨团队协作能力,能够独立负责复杂技术任务。
加分项:
有Atlas系列产品、大模型推理、多卡并行、CUDA模型迁移或边缘AI产品落地经验。