职位详情
【暑期实习】AI性能优化工程师
面议
北京达佳互联信息技术有限公司
北京
本科
04-02
工作地址

快手总部

职位描述
职位描述
1.参与AI与GPU相关项目的性能优化与研发,通过利用CPU、GPU的并行计算优化、架构优化、量化优化和异构调度等高性能优化技术,研发行业领先的高性能异构AI优化技术与编译优化技术;
2.针对搜索、推荐、广告、音视频以及大模型场景,优化模型训练和推理场景的性能;
3.与公司各算法部门深度合作,对重点项目进行算法与系统的联合优化。
任职要求
1.本科及以上学历,计算机相关专业;
2.在图优化、量化、算子优化等技术方向其中一项有深入研究;
3.熟练掌握C++编程语言,具备扎实的数据结构与算法能力,熟悉计算机体系结构和X86汇编,熟悉Python编程;
4.熟悉XLA、MLIR、TVM、Triton、TensorRT等技术之一,并有相应开发经验者优先;
5.熟悉CPU(ARM/x86)或GPU(Intel/Nvidia/AMD)平台的高性能计算优化技术,对计算机体系结构有深入理解,熟悉并行计算优化、访存优化和低比特计算等;
6.了解深度学习算法基本原理,熟悉神经网络基本架构及其算子计算方式,了解至少一种深度学习训练框架及其模型文件解析,如Pytorch、TensorFlow;
7.有使用GPU做AI算法加速相关经历,熟悉GPU CUDA编程;
8.具有独立解决问题的能力,能够对业务逻辑进行合理的抽象和拆分,具备良好的团队合作精神;
9.了解主流AIGC算法模型原理,具有AIGC模型加速优化经验者优先。

以担保或任何理由索取财物,扣押证照,均涉嫌违法,请提高警惕

在线网申