岗位描述(节选)
1. GPU算子库和通信库开发和优化,如CUTLASS等计算库,NCCL通信库。
2. 熟练使用性能分析工具、模拟器和测试套件对软件性能进行分析、可视化与优化。
3. 积极与相关团队合作,交付高度优化的计算/通信库,提升模型推理的整体性能。
【任职要求】
1. 计算机科学、电子工程或计算机工程相关专业硕士及以上学历,至少3年以上相关经验。
2. 深厚的优化Kernel开发背景(如 CUTLASS),具备数学库性能分析和性能剖析的工作经验,能够识别性能瓶颈。
3. 熟练掌握 Python、C、C++ 等编程语言。