职位描述
1. 面向国内外多种 GPU 或其他 AI 算力进行高性能算子开发与优化。
2. 面向实际模型与实际硬件平台,进行算子与推理引擎的综合优化。任职要求
1. 计算机科学或相关领域硕士研究生及以上学历。
2. 具有良好的学习能力和团队合作精神。
3. 掌握 Python、C++、C 等通用编程语言,以及 CUDA 或 AscendC 等算子编程语言。
4. 具备 GPU 等加速卡的算子优化或算子自动编译优化经验。
5. 具备量化矩阵乘法、Flash Attention、计算通信融合算子等复杂算子优化经验者优先。
6. 熟悉 PyTorch 框架、熟悉大模型完整推理流程者、熟悉算子与推理引擎联合优化者优先。
7. 熟悉国内外最新 AI 算力硬件体系结构及其优化方法者优先。