立即沟通
孙先生
感兴趣

BOSS直聘猎头·猎头顾问

50-80K·15薪

深度学习框架高级研发工程师 猎头职位

代招公司:某大型知名互联网公司 不需要融资

北京经验不限本科

立即沟通
孙先生
感兴趣

BOSS直聘猎头·猎头顾问

职位描述

(大模型预训练方向)
岗位职责
1.参与开发优化大模型训练框架,支持单任务万卡以上规模高效稳定训练;
2.参与NLP、多模态大模型结构设计,并联合业务进行模型训练效率和效果验证;
3.参与文生图、文生视频、文生3D等业务的训练性能加速
4.参与低精度训练性能优化和业务推广、参与大窗口训练性能优化
岗位要求
1.熟练使用PyTorch框架,可对DDP训练的代码进行性能分析和优化。
2.熟练使用主流大模型训练框架DeepSpeed、Megatron,掌握3D并行、ZeRO机制、Flash-Attn等的原理、使用场景、优劣势以及可优化方向。
3.有ViT、SD、DiT模型训练性能优化经验者优先。
4.熟练掌握CUDA性能优化手段,有算子编写优化项目经验者优先。
5.对大模型前沿技术比较敏锐者优先。
6.有实际大模型的训练调参和效果评测项目经验的优先;
7.良好的沟通能力、解决问题能力。
更新于:2025-01-21
下载APP确定

已安装?在APP中 直接打开