大模型优化工程师
岗位摘要
分布式训练框架开发:深度优化框架在性能、内存、并行策略等方面的表现,确保大规模训练场景下的效率与稳定性;大模型训练解决方案构建:针对大语言模型及视觉生成等场景,构建覆盖完整训练流程的解决方案
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 分布式训练框架开发:深度优化框架在性能、内存、并行策略等方面的表现,确保大规模训练场景下的效率与稳定性
- 大模型训练解决方案构建:针对大语言模型及视觉生成等场景,构建覆盖完整训练流程的解决方案
- 数据预处理、模型训练、结果评估及训练工具的全方位优化
- 前沿技术追踪与融合:密切关注业界最新技术动态,对前沿技术进行深度研究与分析,推动框架持续迭代升级
任职要求
- 本科及以上学历,计算机相关专业
- 熟悉大模型训练框架Megatron-LM/Deepspeed等
- 具备良好的编程能力,熟练掌握性能分析工具和方法,熟悉PyTorch性能分析工具
- 逻辑思维清晰,具有良好的沟通能力、抗压能力和责任心
- 具有相关开发或优化经验者优先
福利待遇
- 具有竞争力的薪酬:30-60K·15薪
- 知名AI芯片企业寒武纪平台,职业发展前景广阔
- 技术氛围浓厚,专注前沿技术研究与落地
工作地址
北京海淀区致真大厦D1601
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。