AI大模型加速优化算法工程师-(A77402)
薪资面议
AI大模型加速优化算法工程师-(A77402) 薪资面议
北京
本科
0
发布于 8月7日
职位描述
PyTorch
工作内容: 1、进行大模型、多模态大模型、AIGC模型、自驾模型等多个方向端侧加速优化算法的研发与探索; 2、工作内容包括但不限于上述模型的训练调优、量化、剪枝、小型化、稀疏优化、投机推理、专用加速、结构搜索(NAS)等; 3、与芯片团队配合针对相关加速特性进行芯片算法联合设计与优化,并提供针对自研芯片的完整算法加速解决方案; 4、帮助客户在自研芯片上进行算法加速优化的模型加速优化训练及交付。 任职要求: 1、对AI算法(尤其是大模型、AIGC、AD算法)的量化剪枝、网络结构搜索、稀疏化、性能效果优化和训练调参等有丰富开发调试经验; 2、精通transfomers,diffuser,pytorch,tensorrt等框架或平台; 3、有大模型/AIGC/AD算法加速优化项目经验者或在相关领域发表过学术论文或有开源项目者优先。
工作地址
北京/北京/海淀区 西二旗中路33号小米移动互联网产业园
相似职位推荐