字节跳动 社会招聘
AIGC-模型训练加速工程师(北上杭深)
北京 ·经验不限·学历不限
北京
薪资面议
前往 字节跳动 官网投递
职位描述
1、负责模型训练优化算法的研究与开发,运用数据并行、模型并行、通信优化等前沿技术,极大提升模型训练速度及效率;
2、开发和优化AI研发场景下的工具和方法,持续改进并行训练模型的框架和策略,以满足公司核心业务的需求;
3、对业务进行性能瓶颈分析,实施模型训练性能优化措施,提升训练效率,同时利用硬件特性最大化资源效用,并与业界技术进步保持同步。
【任职要求】
1、拥有计算机科学或相关领域的硕士及以上学位,对机器学习、深度学习、计算机视觉、自然语言处理等技术有深刻理解;
2、具备坚实的计算机视觉和自然语言处理基础,熟悉LLM/VLM/SD等主流深度学习模型,并有实际模型训练经验;
3、精通Python编程语言,熟悉Pytorch、FSDP、DeepSpeed、Megatron等分布式训练框架,优先考虑有优化大规模预训练模型经验的候选人。
官网发布:2023-12-20 · 最后确认在招:2026-09-03 19:35:21 · 来源平台:feishu