推理研发实习生

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

岗位描述：

1、参与自研 AI 编译框架（类 TVM/MLIR）的核心研发与架构优化。

2、实现计算图 IR 及图优化 Pass（算子融合、内存复用、并行调度等）。

3、开发多后端代码生成与自动调优能力（如 AutoTVM/Ansor）。

4、负责主流框架（PyTorch/ONNX 等）模型转换与部署，支持 LLM 编译优化。

任职要求：

1、2028届本科及以上学历在校生，计算机、电子、自动化等相关专业。

2、熟练掌握 C++/Python，具备扎实的编程功底。

3、熟悉计算机体系结构、并行编程及内存模型。

4、了解编译器原理，对图优化技术或 Pass 机制有一定认知。

5、具备良好的逻辑思维能力和团队协作精神。

加分项：

1、接触过 TVM、MLIR 或 XLA 等主流 AI 编译框架。

2、了解自动调优技术，或有 CUDA/OpenCL 并行编程经历。

3、熟悉常见深度学习模型（如 CNN/Transformer），对推理优化有一定了解。