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

岗位职责
1.AI工具链架构设计，主要负责大模型训练方向的关键技术。
2.负责训练框架、编译器、算子等模型训练上下游的方案讨论和设计。
3.输出技术架构方案，指导开发团队的工作。

岗位要求
1.计算机、电子等相关专业硕士以上学历。
2.熟悉Megatron/DeepSpeed/Pytorch等训练框架，并且有深入的训练框架开发经验。
3.熟悉AI图编译、算子高性能优化等相关技术。
4.熟悉大模型多机多卡训练的各种并行优化技术。
5.具备丰富的大模型多机多卡集群训练精度调试经验。
6.具备丰富的大模型训练性能和存储占用调优经验。
具有较强的解决问题能力，良好的沟通以及协调能力，具有敬业精神。