深度学习编译实习生(工程岗)
200-500元/天
深度学习 北京 本科 5天/周 最少3个月 有转正
岗位关键词
岗位职责
参与研发业界领先的深度学习编译技术,落地计算优化、显存优化及分布式优化技术到训练框架和推理框架中,赋能深度学习算法落地。
工作内容包括但不限于
1、XLA 相关编译优化功能开发:自动圈图,支持动态shape等
2、参与编译工具链的开发完善相关工作,打造平台化的 AI 编译基础设施。
岗位要求
1、全日制在京实习(不接受远程);
2、了解AI infra 整体技术栈需求,有训练框架或推理框架实战经验、熟悉Tensorflow 或 PyTorch 的使用、有二次开发能力或开源社区贡献经历更佳;
3、(加分项)有大模型相关训练或推理优化经验或GPU 高性能算子开发经验;
4、(加分项)熟悉 深度学习编译优化或异构硬件,有 XLA/ TVM /MLIR 开发、优化经验,熟悉pass编写或代码生成原理和实践;或有传统编译器开发经验,熟悉LLVM原理和使用;5、至少保障3个月实习期,不接受中途学校有事停止实习者; 优先长期实习;
牛客安全提示:如发现虚假招聘、广告,或以任何名义索要证件、费用,或诱导异地入职、参与培训等均属违法行为,发现请
立即举报