欢迎加入摩尔线程!我们团队聚焦主流深度学习框架、大模型及传统模型的完整生命周期,致力于为 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 等深度学习编译器
有模型性能调优、模型压缩、算子开发相关项目实践