探索AI框架前沿｜OneFlow招聘深度学习研发工程师（实习）

一、岗位名称：深度学习研发工程师-框架开发方向（实习）

岗位职责

1. 参与 OneFlow 框架开发、重构与性能优化；

2. 参与深度学习编译、高阶自动微分等深度学习框架相关技术演进工作。

岗位要求

1. 计算机或电子通信相关专业，本科及以上学历；

2. 具备C/C++、Python编程基础，有良好的软件开发素养，熟悉TDD、CI/CD、敏捷开发流程等；

3. 了解深度学习模型，有一定机器学习基础；

4. 熟悉主流开源深度学习框架（如PyTorch、TensorFlow等）内部实现者优先；

5. 熟悉深度学习编译优化技术（如TVM/MLIR/IREE/Triton等）优先。

二、岗位名称：深度学习研发实习生-硬件适配方向（实习）

岗位职责

1.参与OneFlow框架与第三方AI芯片的适配开发工作；

2.参与异构硬件上高性能算子开发、调优、功能测试、性能测试工作。

岗位要求

1. 计算机或电子通信相关专业，本科及以上学历；

2. 具备C/C++、Python编程基础，熟悉Linux系统常用命令、Git代码协作开发的一般流程；

3. 掌握深度学习基本原理，熟悉基于AI框架进行模型训练推理的一般流程；

4. 有基于主流开源深度学习框架进行硬件适配、算子开发、性能调优相关经验者优先；

5.有NVIDIA CUDA、AMD ROCM，熟悉GPU（CUDA、cuDNN、NCCL）编程者优先；

6. 有昆仑芯XPU、寒武纪MLU、昇腾NPU开发/优化经验者优先。

简历发送地址：talent@oneflow.org