中科南京软件技术研究院大模型算法工程师招聘启事

【文章来源:人力资源处】【发布时间:2024-01-22阅读次数:2293】【我要打印

      中科南京软件技术研究院成立于2019年8月,是南京市登记设立科研事业单位(新型研发机构),主要面向计算机科学和软件领域开展基础研究和应用基础研究、应用示范及产业化。

      现因科研业务发展需要,诚邀有志从事大模型算法研究的年轻人士加入团队,共同发展,共筑辉煌。

      一、招聘岗位

      大模型算法工程师  1-2名

      大模型算法工程师实习生 1-2名

      二、岗位职责

      1、开展预训练模型训练加速相关研究,包括但不限于优化PyTorch、Tensorflow等深度学习框架,使用Megatron、DeepSpeed等框架对预训练模型进行优化,进行大模型量化计算、网络通信、算子融合优化;

      2、负责预训练模型的部分加速,包括但不限于多卡任务调度,算子优化等;

      3、与团队成员协同工作,设计实现各类模块、功能,并进行性能调优,解决团队研发过程中遇到的效率问题,如分布式计算加速、性能瓶颈分析和评测、模型算法优化等;

      4、探索创新,积极探索新方法、不断提升工作效率及团队协调开发能力。

      三、任职要求

      1、硕士及以上学历,计算机系统结构专业优先,计算机科学与技术/软件工程/人工智能/大数据/电子信息/通信等相关专业优先;

      2、良好的编程风格与文档撰写能力,有多人协作开发及Code Review经验,对代码有严格的质量要求;熟悉常见开发规范,熟悉Git等开发工具,熟悉代码库的创建与维护操作;

      3、熟悉GPU架构、熟悉国产自研加速芯片架构的优先,如华为异腾、百度昆仑、寒武纪、DCU等;熟悉CUDA编程,有一定的高性能编程经验,有神经网络训练加速、推理加速和显存优化经验的优先;掌握深度学习基础知识,至少熟悉一种深度学习编程框架,如PyTorch、Tensorflow、Paddle、MindSpore等,有预训练模型开发、部署、调试经验的优先;有计算优化加速经验软件/体系结构开发经验、系统级调优经验的优先;

      4、工作常驻地为北京、南京。

      备注:提供实习岗位,欢迎特别优秀的本科生、研究生实习。

      四、薪酬待遇

      提供具有高竞争力的人事、工资、保险等福利待遇。

      五、应聘方式

      有意竞聘者,请将个人简历发送至nanjinghr@njis.ac.cn,请在邮件标题中注明应聘岗位名称。

      此招聘启事长期有效,至招聘人员到岗为止。