AI 工程师（算子精度）

负责研究大模型训练过程中算子的数值精度特性与误差传播规律，建立训练精度评估体系，为低精度训练、Kernel 优化、自动化算子搜索等方向提供理论依据与实验支撑。参与下一代训练系统中 Kernel Evolve Agent 等自动优化系统建设。

岗位职责

研究训练过程中各类算子的数值稳定性与精度敏感性。

分析算子误差对训练收敛性、模型能力和最终性能的影响。

建立算子级精度评估与测试框架。

设计自动化实验方法，评估不同精度配置下的训练效果。

参与 Kernel 优化与自动搜索系统建设，为算子精度约束提供评价指标。

与预训练、后训练团队协作推进相关研究。

任职要求

计算机、数学、统计、人工智能等相关专业本科及以上学历。

扎实的线性代数、数值计算、概率统计基础。

熟悉深度学习训练过程及优化算法。

熟练掌握 Python，具备较强实验分析能力。

能够独立设计实验并分析结果。

加分项

具有低精度训练经验。

具有 TileLang、Triton、CUTLASS、CuteDSL、CUDA 等算子开发精度对齐经验。

具有数值分析、误差传播相关研究背景。

具有 Compiler、Kernel Optimization 相关研究经验。