AI框架适配优化研发工程师
飞腾信息技术有限公司岗位职责:1. 基于计算设备编程框架,开发与主流深度学习框架兼容的算子; 2. 基于计算设备底层软件栈,对主流深度学习框架进行适配开发; 3. 分析主流深度学习框架在计算设备上的性能瓶颈,并进行性能优化; 4. 进行主流大模型并行训练框架的适配与优化。 岗位要求:1. 熟练掌握深度学习算法,能熟练地训练神经网络模型; 2. 熟悉主流框架中的常用算子及其数学意义,有实现经验者优先; 3. 熟悉主流深度学习框架,包括但不限于 PyTorch、TensorFlow、JAX、PaddlePaddle、MindSpore 等,对其中一种或多种的底层实现有较深入研究者优先; 4. 熟悉并行训练框架,了解大模型训练或微调技术,有实…