应届毕业生
本科
2025-08-05 10:35:50
7062人关注
职位描述
该职位已进行加V认证，请放心投递
职位描述
基于NPU平台，开发高性能计算库，支撑CV，NLP，大模型等模型运行，包括：
1.完成高性能算子的需求分析与分解，满足NPU的各种应用场景；
2.与芯片/架构团队紧密配合，持续优化算子性能，逐步完善软硬件架构；
3.完成各类算子的软件方案设计及开发；
4.分析算子/网络的性能瓶颈，提出性能优化策略，达成性能目标；
职位要求
1.图像处理、模式识别、机器学习等相关专业硕士及以上学历；
2.精通C /Python软件编程语言；
3.熟悉Onnx/TensorFLow/Pytorch一种主流神经网络框架，熟悉各类算子；
4.熟悉并行/异构计算体系结构，在TPU/NPU/GPU一种平台有2年以上的开发和优化经验；
5.具备业界常用高性能库(TensorRT/cuDNN/CUDA/OPENCV)开发和性能优化经验者优先；
6.熟悉指令流水，计算机体系结构；
7.良好的团队沟通能力、技术攻关能力，责任心强，具备一定抗压能力。
注：联系我时，请说是在白云人才网上看到的。
工作地点
地址：北京海淀区北京-海淀区海淀区
求职提示：用人单位发布虚假招聘信息，或以任何名义向求职者收取财物（如体检费、置装费、押金、服装费、培训费、身份证、毕业证等），均涉嫌违法，请求职者务必提高警惕。
职位发布者
HR
小米科技有限责任公司
-
通信/电信/网络设备/增值服务
-
1000人以上
-
私营·民营企业
-
北京市海淀区清河中街68号华润五彩城写字楼
