AI算子性能优化工程师
岗位摘要
面向GPU/昇腾架构开展算子性能画像、热点定位与优化,持续提升吞吐与时延稳定性;结合训练与推理场景,协同框架与编译链路进行内核调优、算子融合、内存访问与并行策略优化;建设性能基线与度量方法,沉淀可复用的优化范式与最佳实践,推动持续优化闭环
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 面向GPU/昇腾架构开展算子性能画像、热点定位与优化,持续提升吞吐与时延稳定性
- 结合训练与推理场景,协同框架与编译链路进行内核调优、算子融合、内存访问与并行策略优化
- 建设性能基线与度量方法,沉淀可复用的优化范式与最佳实践,推动持续优化闭环
任职要求
- 理解计算图、并行化、向量化、内存层次与缓存友好性等关键要点
- 熟悉主流AI算法与框架(如Transformers、PyTorch、TensorFlow等)
- 具备GPU或NPU底层优化经验,熟悉CUDA或CANN编程模型
- 掌握性能分析工具(如Nsight、Profiler等)的使用方法
- 具有扎实的C++/Python编程能力和算法基础
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。