深度学习框架研发工程师

15-30K

长沙市不限学历不限经验

职位描述

工作职责：

1. 基于计算设备编程框架，开发与主流深度学习框架兼容的算子2. 基于计算设备底层软件栈，对主流深度学习框架进行适配开发3. 分析主流深度学习框架在计算设备上的性能瓶颈，并进行性能优化4. 主流大模型并行训练框架适配与优化任职要求：

1. 熟悉掌握深度学习算法，能熟练地训练神经网络模型2. 熟悉主流框架中的常用算子及其数学意义，有实现经验的优先3. 熟悉主流深度学习框架，包括但不限于Pytorch、Tensorflow、Jax、PaddlePaddle、MindSpore等，对其中之一或多个的底层实现有较深入研究的优先4. 熟悉并行训练框架，了解大模型训练或微调技术，有实际项目经验的优先5. 熟悉CUDA、OpenCL等异构编程框架的优先；熟悉CUDA软件栈、oneAPI软件栈或ROCm软件栈的优先6. 熟练掌握Python和C/C++，了解Python与C/C++相互调用方法；熟悉Linux，熟练掌握Shell

登录查看完整内容

20,861+ 岗位更新等你来订阅

一键订阅最新的岗位，每周送达

您可以在邮箱中随时取消订阅