2024中科南京软件技术研究院大模型算法工程师岗位职责

　　招聘岗位

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

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

　　岗位职责

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

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

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

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

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