岗位描述
职位描述:
负责算子性能手动/编译优化;
调研前沿论文/技术;
参与基础性工程开发,包括:
1)常态化负责厂商测试管理、脚本维护与更新,完成例行任务分发,更新数据和指标;
2) 阶段性负责算子库、编译器项目测例补充,版本依赖管理,性能看板建设,严格把控交付质量;
3) 阶段性负责算子库、编译器项目的基础设施建设;
任职要求:
本科及以上学历,计算机或相关专业优先, 具备较为扎实的编程理论功底;
熟悉 Python、C++ 等编程语言,Shell、CMake 等算法工程化技术及其相关工具链,具备基本的脚本编程、数据分析能力;
了解 TensorFlow、Pytorch、PaddlePaddle 等一种或多种深度学习框架;
了解大模型训练/推理基本原理及流程;
具备快速的 case by case 学习能力,具备对比分析和逻辑推断能力,对技术有热情,有责任心,有良好的团队合作和协调沟通能力。
加分项:
熟悉以 Triton、CUDA 等算子开发范式为中心的算子编写技术;
熟悉以 TVM、LLVM、MLIR 等范式为中心的AI编译器开发技术。