职位描述
![](http://img.jrzp.com/jrzpfile/provincercw/images/fangxintou.png)
职位描述
基于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.良好的团队沟通能力、技术攻关能力,责任心强,具备一定抗压能力。
工作地点
地址:北京海淀区北京-海淀区海淀区
![](http://img.jrzp.com/jrzpfile/rcw/SearchJob/images/jg.png)
![](https://img.jrzp.com/images_server/comm/nan.png)
职位发布者
HR
小米科技有限责任公司
![](http://img.jrzp.com/jrzpfile/provincercw/images/sfrz_yrz.png)
![](http://img.jrzp.com/jrzpfile/rcw/SearchJob/images/V.png)
-
通信/电信/网络设备/增值服务
-
1000人以上
-
私营·民营企业
-
北京市海淀区清河中街68号华润五彩城写字楼