岗位描述
岗位职责
1. 负责AI编译器设计与开发:前端解析、IR优化(图优化/算子融合/常量折叠)、后端代码生成与自动调优,构建兼容主流生态(PyTorch/Triton/CUDA算子语义)的编译能力;
2. 负责大模型(Transformer/MoE)高性能算子(Attention/GEMM等)的研发与调优(CUDA/Triton/CUTLASS);
3. 负责深度学习框架(PyTorch)适配与分布式编译;
4. 协同硬件团队挖掘芯片算力极限,保障大模型训推性能。
岗位要求
1. 本科及以上,计算机/电子相关专业;专家岗5年以上、开发岗3年以上编译器或算子开发经验;
2. 精通C/C++/Python,熟悉LLVM/TVM/MLIR/Triton等编译框架;
3. 熟悉GPU/NPU架构,精通CUDA编程与性能调优工具(Nsight);
4. 有大模型推理框架(vLLM/SGLang/TensorRT-LLM)实战经验,或LLVM/MLIR/TVM社区Committer经历者优先。