职位描述
1. 研究和优化Triton编译器,使其高效支持芯片架构;
2. 基于MLIR + Linalg框架,提升编译器代码执行效率;
3. 研究算子融合、自动调优、代码生成等相关优化策略;
4. 与硬件团队、AI框架团队(如PyTorch/TensorFlow) 协作,提升AI编译器的支持能力;
5. 跟踪LLVM、Triton、TVM、XLA、TorchInductor等AI编译器前沿技术,优化编译性能。任职要求
1. 计算机、微电子、电子工程、软件工程等相关专业,硕士及以上学历,对AI编译器、AI计算优化有浓厚兴趣;
2. 熟悉C++/Python,具备编译器或 AI 编译优化开发经验,熟悉 MLIR、Linalg等基础设施;
3. 在PLDI、CGO、MICRO、MLSys、ISCA等学术会议上发表过AI编译相关论文为加分项。
4. 有以上相关的实习/项目经历为加分项;
5. 有良好的沟通能力和团队协作能力,优秀的学习能力和技术热情。