岗位描述
职位描述:
1.负责量子计算模拟算子性能手动/编译优化;
2.调研前沿论文/技术;
3.参与基础性工程开发,包括:
1)基于TorchQuantum等框架、Triton语言在国产GPU上实现量子计算模拟,赋能量子计算;
2) 在指导下参加算子库、编译器项目的基础设施建设;
任职要求:
1.本科及以上学历,计算机或相关专业优先, 具备较为扎实的编程理论功底;
2.熟悉 Python、C++ 等编程语言,Shell、CMake 等算法工程化技术及其相关工具链,具备基本的脚本编程、数据分析能力;
3.了解 TensorFlow、Pytorch、PaddlePaddle 等一种或多种深度学习框架;
4.了解大模型训练/推理基本原理及流程;
5.具备快速的 case by case 学习能力,具备对比分析和逻辑推断能力,对技术有热情,有责任心,有良好的团队合作和协调沟通能力。
加分项:
1.具有量子计算相关专业背景;
2.对大模型推理工程化实践具有浓厚的兴趣;
3.熟悉以 Triton、CUDA 等算子开发范式为中心的算子编写技术;
4.熟悉以 TVM、LLVM、MLIR 等范式为中心的AI编译器开发技术。