【HOT岗位】工具链架构师（训练方向）

1.AI工具链架构设计，主要负责大模型训练方向的关键技术。

2.负责训练框架、编译器、算子等模型训练上下游的方案讨论和设计。

3.输出技术架构方案，指导开发团队的工作。

1.计算机、电子等相关专业硕士以上学历。

2.熟悉Megatron/DeepSpeed/Pytorch等训练框架，并且有深入的训练框架开发经验。

3.熟悉AI图编译、算子高性能优化等相关技术。

4.熟悉大模型多机多卡训练的各种并行优化技术。

5.具备丰富的大模型多机多卡集群训练精度调试经验。

6.具备丰富的大模型训练性能和存储占用调优经验。

具有较强的解决问题能力，良好的沟通以及协调能力，具有敬业精神。