logo of cambricon

寒武纪大模型训练框架研发工程师-27届

校招全职地点:北京状态:招聘

工作描述


岗位职责:
参与AI大模型训练框架的研发适配工作,分析优化性能热点,分析内存瓶颈,优化大模型训练框架;
调研分析大模型最新技术和性能优化方案,跟进社区演进方向,持续优化大模型训练瓶颈问题;
性能分析的理论计算及预估,对计算、通信时间和内存占比能够进行合理性分析,并提出优化方案;
充分理解分布式训练全过程,优化预训练及后训练阶段性能,不断提高…
登录查看完整工作描述
微信扫码,1秒登录

包括英文材料
大模型+
学历+
PyTorch+
Linux+
Unix+
还有更多 •••
相关职位

logo of xiaohongshu
社招5-10年引擎

任职资格: 精通 PyTorch,具备训练框架源码级阅读与修改能力,有实际性能优化经验。 熟练掌握 Megatron、DeepSpeed、veRL、OpenRLHF、TRL、Llama-Factory

更新于 2026-08-02上海|北京|杭州
logo of kuaishou
社招3-5年J0012

1、具备良好的团队协作能力,热爱钻研技术,善于分析、解决工程问题; 2、有深度学习基本知识,熟悉Transformer/Bert/GPT等模型结构更佳; 3、掌握Python/C++编程语言,熟练使用

更新于 2026-06-01北京|上海
logo of tencent
社招3年以上腾讯云技术

1.计算机科学、数学、电子工程等相关专业硕士及以上学历,3年以上深度学习框架或工具链开发经验,精通Python,熟练使用C++; 2.有Megatron-LM、DeepSpeed等开源框架使用或二次开

更新于 2026-07-29深圳
logo of tencent
社招3年以上腾讯云技术

1.计算机科学、数学、电子工程等相关专业硕士及以上学历,3年以上深度学习框架或工具链开发经验,精通Python,熟练使用C++; 2.有Megatron-LM、DeepSpeed等开源框架使用或二次开

更新于 2026-07-27杭州