職位描述

職位描述
基于AI芯片,完成AI推理框架的方案設計及開發,包括:
1.基于AI芯片,完成 AI 推理框架的開發和交付,適配CPU/GPU/NPU多后端,支持CV/NLP/大模型等端側落地;
2.負責框架與FrameWork的對接,包括AndroidNN/Tensorflow/Pytorch/Onnx等;
3.負責框架相關工具鏈的開發,包括Memory/PerformanceProfiling,量化工具等;
4.負責CV/NLP/大模型等算法業務的端到端性能優化,滿足商用交付需求
職位要求
1.碩士及以上學歷,計算機/電子/信息/通信/自動化/軟件等相關專業,有AI工程優化經驗的優先;
2.精通C/C ,精通算法與數據結構,熟悉Python;
3.深入了解PADDLE/MNN/TVM/MLIR等框架中的一種或多種,熟悉算子性能優化、算子融合、graph遍歷等技術;
4.熟悉NPU或其他AI芯片工具鏈開發, 熟悉模型量化技術優先;
5.了解主流 AIGC 算法模型原理,有AIGC模型加速優化經驗者優先。
6.熟悉Android NN/TensorFLow/Pytorch/Caffe中一種神經網絡框架;
工作地點
地址:上海浦東新區上海-浦東新區5G未來中心


職位發布者
HR
小米科技有限責任公司


-
通信/電信/網絡設備/增值服務
-
1000人以上
-
私營·民營企業
-
北京市海淀區清河中街68號華潤五彩城寫字樓