高性能计算工程师

工作职责:

1、负责开发面向大模型需求的高性能算子库,优化kernel性能,达到业界领先

2、负责算子库的大模型集成验证,达到端到端训练的精度和性能要求

任职要求:

1、计算机或相关专业硕士及以上学历

2、具备GPU高性能算子开发经验,精通CUDA高性能编程者优先

2、具备扎实的C++、Python编程基本功和工程开发经验

3、熟悉深度学习系统或有大模型训练底层开发相关经验

4、对技术有热情,有责任心,有良好的团队合作和沟通能力

公司地点:北京海淀区智源大厦智源大厦

公司简介:

职位发布者:杜女士

北京智源人工智能研究院

融资阶段:

公司规模:

相似职位: