返回岗位列表

AI基础设施研发工程师

查看原岗位

地点：北京薪资：30-60K 日期：2026-03-03

岗位摘要

参与PyTorch、PaddlePaddle与摩尔线程软硬件栈的对接、训练、推理与部署；实现推理框架及深度学习算子的性能优化；研究并实现单GPU与分布式训练技术，包括并行与异构计算；针对自研GPU设计模型优化与压缩方案

岗位职责

参与PyTorch、PaddlePaddle与摩尔线程软硬件栈的对接、训练、推理与部署
实现推理框架及深度学习算子的性能优化
研究并实现单GPU与分布式训练技术，包括并行与异构计算
针对自研GPU设计模型优化与压缩方案

任职要求

熟悉Linux、C++、Python，算法扎实，编程规范，系统设计能力强
深入理解CV/NLP算法，有主流框架开发或优化经验
具备XLA、TVM等深度学习编译栈开发经验优先
精通分布式训练，熟悉参数服务器、多维并行、显存交换等技术

福利待遇

16薪高薪体系
前沿GPU与AI技术实战机会
望京核心办公，团队技术氛围浓厚

工作地址

北京朝阳区望京国际研发园l座3层