【星火X计划】面向大模型高效训推的算子研发与极致优化(J14504)

地点

合肥、北京

团队

词元星火

雇佣形式

全职

招聘类型

校招

尚未披露：薪资、雇主申请截止时间

投递在官方渠道完成，完整外部回执留在本人工作区；已登录时，通过 CLI 明确记录的选择、本人自报投递和渠道观察三类最小事件默认上报平台。浏览或保存岗位不会触发上报，外部投递不会出现在“我的申请”里。

• 本课题专注于对大模型训推过程中的关键算子进行极致效率优化
• 实现高算力利用率的高性能Kernel开发
• 结合超节点等互联架构，实现计算与通信高效融合

任职要求与期待

• 24-27届硕士及以上学历，人工智能、计算机科学、高性能计算、数学、电子信息等相关专业，其他专业但具备突出工程能力者同样欢迎
• 具备扎实的计算机架构基础，熟练使用linux操作系统
• 熟练使用Git、VSCode等，熟练使用Pytorch、numpy等框架
• 对GPU/NPU等AI加速芯片的体系结构有深入理解，熟悉CUDA/Ascend C/Triton/TileLang等至少一种编程模型，有实际的Kernel开发调优经验
• 较强的编程实现能力，熟练掌握C/C++，Python等语言，有底层性能优化项目经验
• 具备较强的独立思考能力，具备追求卓越，不断突破创新精神，不断通过效率分析提升算子效率。