AI算法工程师
1.2-2万·13薪上海市本科不限经验
职位描述
具有其中一方面即可
* 通信库开发
负责AI芯片的软件栈通信库的移植,开发和调试和测试
任职要求:
1. 有AI芯片通信库(NCCL 或者HCLL等开源通信库)移植和设计、开发经验优先
2. 有熟悉算力中心网络开发经验优先
3. 熟悉以太网,RDMA协议和TCP/IP协议其中之一
4. 熟悉Linux 环境编程
5. 精通常用的数据结构算法和编程
6. 很强的C/C++开发能力
7. 有软硬配合调试和性能分析经验优先
*AI算子库开发
算子(Operator):基于公司编程模型,在自研AI芯片上开发实现各种AI算子。
在功能上泛化支持完整算子语义,包括但不限于任意算子形状和数据类型的支持;
在性能上能够将芯片算力发挥;
在算子融合上,能够高效率支持开源框架定义的通用算子融合场景。
职位要求:
有扎实的C/C++/Python编程能力
良好的算法基础,熟悉时间复杂度和空间复杂度的计算方法
软件Debug能力出众,能够高效的定位bug范围。
NPU/GPU上底层函数库的功能开发和性能优化经历者优先
有大模型flashattention开发经验优先
* 深度学习框架开发
岗位要求:
1,有Python的编程能力
2、熟悉C/C++,良好的编程习惯,熟悉计算机基础算法和数据结构;熟悉linux开发环境;
3、有深度学习框架(Pytorch,Tensorflow,PaddlePaddle)开发和使用经验;
4、熟悉MPI,有大规模训练调优经验优先,有深度学习算子开发经验有限;
20,861+ 岗位更新等你来订阅
一键订阅最新的岗位,每周送达
🎉恭喜你,订阅成功
继续订阅您可以在邮箱中随时取消订阅