算子开发工程师(PyTorch)

发布时间: 2025-08-01 20:30 工作地点: 北京 薪资: 30-50万

必备条件:
必须是一本学历,计算机相关专业,经验1-3年,年龄超38岁不考虑
优先条件:
暂无
岗位职责:
职位描述
1、面向国内外多种 gpu 进行高性能算子开发与优化.
2、面向在不同 gpu 间移植高性能算子的需求,参与编译技术开发
任职要求:
职位要求
1、计算机科学或相关领域硕士研究生及以上学历
2、熟悉 python、c++、c 等编程语言
3、具备编译优化经验或 gpu 等加速卡的算子优化经验
4、熟悉深度学习、熟悉 pytorch 等深度学习框架者优先
5、具备矩阵乘、flash attention 等计算密集型算子优化经验者优先
6、熟悉 triton、tvm、mlir 等深度学习专用编译器或编译器组件等的优先
7、有良好的沟通和协调能力,具备团队合作精神,能够承受⼯作压力,对技术有强烈的热情和求知欲

简历提交评估