工作职责:
1、负责开发面向大模型需求的高性能算子库,优化kernel性能,达到业界领先
2、负责算子库的大模型集成验证,达到端到端训练的精度和性能要求
任职要求:
1、计算机或相关专业硕士及以上学历
2、具备GPU高性能算子开发经验,精通CUDA高性能编程者优先
2、具备扎实的C++、Python编程基本功和工程开发经验
3、熟悉深度学习系统或有大模型训练底层开发相关经验
4、对技术有热情,有责任心,有良好的团队合作和沟通能力