AI软件架构工程师 - AI编译方向 （27届校招）

太初元碁是国家级专精特新“小巨人”与国家高新技术企业，核心团队成员始脱于国家超级计算无锡中心与清华大学，是国内AI算力底座的坚定探索者，坚持走异构众核的差异化技术路线，核心骨干多次荣获“戈登·贝尔奖”。首款自研AI加速卡T100已成功量产并规模商用，全面适配DeepSeek、Qwen（通义千问）、GLM（智谱）等多个主流AI大模型，用真实算力驱动AI未来。

在太初架构上建设和优化 Triton 编译器，提升 AI 计算的性能和效率。

岗位职责

1.在太初AI架构上适配和优化 Triton 编译器；

2.基于 MLIR + Linalg 框架，提升编译器代码执行效率；

3.参与 图优化、后端代码生成、调度策略、指令优化、流水并行化 相关优化工作；

4.研究和优化 Kernel 计算性能，降低计算开销，提高吞吐率；

5.研究算子融合（Op Fusion）、自动调优（Auto-Tuning）、代码生成（CodeGen） 相关优化策略；

6.与 硬件团队、AI框架团队（如 PyTorch） 协作，提升AI编译器的支持能力；

7.跟踪 LLVM、Triton、TVM、XLA、TorchInductor 等 AI 编译器前沿技术，优化编译性能。

任职要求

1.硕士及以上学历，优秀的学习能力，对 AI 编译器、AI计算优化有浓厚兴趣；

2.熟悉 C++/Python，具备编译器或 AI 编译优化开发经验，熟悉 Triton 编译流程；

3.熟悉 MLIR、Linalg等基础设施；

4.具备ai编译优化（如 TVM、XLA、TorchInductor） 经验者优先；

5.熟悉 Kernel 计算优化、算子融合、自动并行化等高性能计算优化策略者加分。

加分项（优先考虑）

1.参与过 Triton、MLIR、IREE、XLA、TVM、TorchInductor 等编译器的开发，或在 GitHub 提交过相关 PR。

2.具备MLU/NPU/GPU 相关的编译优化经验。