岗位描述
岗位职责
1、编译工具链与多层次IR基础设施:构建并维护基于 LLVM 后端、结合 Triton/TileLang的算子编译栈与多层次IR体系;设计高层算子表示与 Kernel IR 的接口规范,打通算子表示、Kernel IR、 后端 IR/LLVM的端到端编译流水线;完善编译器工程化能力;
2、算子与图级性能优化:面向关键算子与子图设计和实现优化策略,包括算子融合与调度、向量化与并行映射、内存访问与布局优化,以吞吐/延迟/显存占用为目标持续迭代;
3、混合精度与量化适配:负责 FP32/FP16/BF16/INT8等精度策略在编译与算子侧的落地,保障训练数值稳定性与精度;与通信库及运行时协同完成端到端验证;
4、A2 后端适配与代码生成:面向 A2 芯片 ISA 与微架构实现后端 lowering 与 codegen,包括指令选择、调度、寄存器与共享内存等资源约束优化,形成高效 kernel 生成与性能调优闭环。
岗位要求
1. 硕士及以上学历,计算机科学与技术、电子工程、人工智能、软件工程等相关专业;
2. 具备 LLVM/MLIR 编译器开发与工程化落地经验,有算子/Kernel 级性能优化项目经验;理解 ISA/微架构与资源约束,具备后端指令选择、调度与资源分配优化经验;参与过 Triton/TileLang 等算子编译/Kernel 生成框架,或在实际大模型训练/推理中进行算子定制与性能调优者优先;有混合精度与量化(FP16/BF16/INT8)相关实践经验者以及有 AI 加速器/GPU/异构体系结构适配经验者优先;
3. 熟悉 LLVM/MLIR 编译生态与多层次IR设计方法,具备 Pass Pipeline 组织与实现能力;熟悉 Triton/TileLang 等 GPU/AI Kernel 生成与调优体系,具备算子级/Kernel级优化经验;熟悉大模型核心算子与计算图执行特征,理解训练/推理中的带宽瓶颈与计算瓶颈,并能将优化策略落到编译与算子实现中;熟悉混合精度与量化相关机制,了解精度传播、溢出/舍入控制、校准与误差约束方法,能与算子库/运行时协同完成端到端落地;
4. 具备优秀的系统性问题分析和解决能力,良好的团队协作与沟通能力,能够承担有挑战性的研发任务,并具备较强的学习能力和创新思维;
5. 对国产AI芯片生态建设有热情,能适应阶段性高强度研发工作;有责任心,注重代码质量与工程规范。