深度学习框架开发工程师(J10982)

职位描述

欢迎加入摩尔线程！我们团队聚焦主流深度学习框架、大模型及传统模型的完整生命周期，致力于为 KUAE 打造高效的万卡集群训练基础设施与业界领先的基座大模型，加入我们，一起探索技术的无限可能！

岗位职责

1、参与主流机器学习框架（如 PyTorch 和 PaddlePaddle）与摩尔线程软硬件栈的对接，开展模型训练、推理和部署相关工作。

2、参与机器学习单 GPU 以及分布式训练技术的研究与实现，聚焦并行与异构计算技术的设计、开发以及性能优化，侧重框架侧适配实现。

3、参与研究并实现适合摩尔线程自研 GPU 的模型优化和压缩方法。

任职要求

任职资格：

1、计算机、软件工程、人工智能等相关专业，具备扎实的算法基础以及系统设计能力，欢迎基础扎实的在读同学投递。

2、熟悉 Linux 开发环境，精通 C++ 和 Python，具备良好编程风格。

3、深入理解 CV 和 NLP 典型深度学习算法，拥有 TensorFlow、PyTorch、MxNet、Caffe、PaddlePaddle 等主流深度框架开发或优化经历。

4、责任心强，学习能力、沟通能力和自驱力突出。

【加分项】

了解 XLA、TVM、MLIR 等深度学习编译器

有模型性能调优、模型压缩、算子开发相关项目实践