性能工程师(预训练团队)
岗位摘要
设计和编写用于训练的高性能、可扩展软件;理解架构修改和设计选择及其对训练吞吐量和质量的影响;编写低层CUDA、Triton内核,以从加速器中榨取每一分性能;在我们的超级计算和数据基础设施上研究、实施和实验各种想法
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 设计和编写用于训练的高性能、可扩展软件
- 理解架构修改和设计选择及其对训练吞吐量和质量的影响
- 编写低层CUDA、Triton内核,以从加速器中榨取每一分性能
- 在我们的超级计算和数据基础设施上研究、实施和实验各种想法
- 向该领域最优秀的研究人员学习并与之合作
任职要求
- 极强的软件工程技能
- 有使用CUDA、Triton等为GPU编写内核的经验
- 有使用大规模分布式训练策略的经验
- 熟悉自回归序列模型,如Transformer
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。