职位名称
AI Infra研发工程师
单位名称
摩尔线程智能科技（北京）股份有限公司
任职要求
1. 熟悉Linux开发环境，熟悉C++和python，具备良好扎实的算法基础、良好的编程风格和系统设计能力； 2. 深入理解典型的深度学习算法（CV、NLP），主流深度学习框架开发或优化经验； 3. 有深度学习编译（XLA、TVM）等软件栈开发经验优先； 4. 精通分布式训练技术，有参数服务器、多维并行、显存交换等技术的深入研究与开发者可优先考虑；
岗位职责
1. 参与主流机器学习框架（如PyTorch、PaddlePaddle）与摩尔线程软硬件栈的对接、模型训练、推理和部署； 2. 参与实现推理框架以及深度学习框架算子的性能优化； 3. 参与机器学习单GPU以及分布式训练技术的研究与实现，如各种并行与异构计算技术的设计、开发以及性能优化； 4. 参与研究并实现适合摩尔线程自研GPU的模型优化和压缩方法；