返回岗位列表

Cohere

性能工程师(预训练团队)

地点:美国 薪资:薪资未公开 日期:2026-01-29

岗位摘要

设计和编写用于训练的高性能、可扩展软件;理解架构修改和设计选择及其对训练吞吐量和质量的影响;编写低层CUDA、Triton内核,以从加速器中榨取每一分性能;在我们的超级计算和数据基础设施上研究、实施和实验各种想法

岗位职责

  • 设计和编写用于训练的高性能、可扩展软件
  • 理解架构修改和设计选择及其对训练吞吐量和质量的影响
  • 编写低层CUDA、Triton内核,以从加速器中榨取每一分性能
  • 在我们的超级计算和数据基础设施上研究、实施和实验各种想法
  • 向该领域最优秀的研究人员学习并与之合作

任职要求

  • 极强的软件工程技能
  • 有使用CUDA、Triton等为GPU编写内核的经验
  • 有使用大规模分布式训练策略的经验
  • 熟悉自回归序列模型,如Transformer