岗位职责:
1、 负责AI训练芯片架构的探索和设计;
2、 负责AI训练系统的软硬件协同设计;
3、 负责不同业务场景下AI模型训练流程中的关键任务和典型算子分析;
1、 熟悉计算机体系结构,对典型AI硬件加速器架构或者GPU架构有深入的了解;
2、 有AI模型优化相关的经验,对典型AI模型计算访存行为有较好的理解;
3、 熟悉分布式训练流程,熟悉方向传播和参数更新过程;
4、 熟练掌握C/C++,python等相关编程语言;
加分项:
1、 计算机网络的TCP/IP协议栈或者RDMA协议有了解;
2、 了解TensorFlow、Pytorch等深度学习框架;