平头哥-边缘AI芯片计算/通信加速库开发工程师-上海

岗位职责
1. GPU算子库和通信库开发和优化，如CUTLASS等计算库，NCCL通信库。
2. 熟练使用性能分析工具、模拟器和测试套件对软件性能进行分析、可视化与优化。
3. 积极与相关团队合作，交付高度优化的计算/通信库，提升模型推理的整体性能。

任职要求
1. 计算机科学、电子工程或计算机工程相关专业硕士及以上学历，至少3年以上相关经验。
2. 深厚的优化Kernel开发背景（如 CUTLASS），具备数学库性能分析和性能剖析的工作经验，能够识别性能瓶颈。
3. 熟练掌握 Python、C、C++ 等编程语言。