职位描述

职位描述
基于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号华润五彩城写字楼
相似职位
-
高级健康管理师 10000-15000元江岸区 应届毕业生 不限湖北优唐健康管理有限责任公司
-
银保渠道经理 面议江岸区 应届毕业生 不限太平人寿保险有限公司武汉中心支公司
-
执业药师/驻店药师 4000-6000元江岸区 应届毕业生 大专国药控股湖北国大药房有限公司
-
基地PQC质量负责人/经理(宁德时代总部) 20000元以上江岸区 应届毕业生 不限宁德时代新能源科技股份有限公司
-
市场/营销专员 6000-19000元江岸区 2年以上 本科武汉力神动力电池系统科技有限公司
-
短期过渡300/天 可预支/月 10000-14000元江岸区 应届毕业生 不限成都泰便利电子商务有限公司