岗位描述
太初元碁是国家级专精特新“小巨人”与国家高新技术企业,核心团队成员始脱于国家超级计算无锡中心与清华大学,是国内AI算力底座的坚定探索者,坚持走异构众核的差异化技术路线,核心骨干多次荣获“戈登·贝尔奖”。首款自研AI加速卡T100已成功量产并规模商用,全面适配DeepSeek、Qwen(通义千问)、GLM(智谱)等多个主流AI大模型,用真实算力驱动AI未来。
岗位描述:
1、参与设计基于 Triton 的端到端编译架构(Triton IR → MLIR Dialect → 后端目标码生成)。
2、定义 compiler pipeline 接口与层级边界,制定 MLIR dialect 划分策略以支撑自研芯片适配。
3、参与 Triton IR 扩展设计,支持硬件特有原语,并研究算子融合等优化策略。
任职要求:
1、2027届硕士及以上学历,计算机相关专业(有实习转正机会)。
2、扎实的 C++ 编程基础,系统学习过编译原理,了解编译器基本工作流程。
3、对 AI 编译技术有浓厚兴趣,接触过 MLIR、LLVM 或 Triton 等框架。
加分项:
1、深入研究或使用过 Triton、MLIR、LLVM、TVM 等编译器基础设施。
2、了解 GPU/NPU 等异构计算架构,或有 CUDA 等并行编程经验。