快手AI性能优化工程师

任职要求

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模型加速优化经验者优先。

工作职责

1.参与AI与GPU相关项目的性能优化与研发，通过利用CPU、GPU的并行计算优化、架构优化、量化优化和异构调度等高性能优化技术，研发行业领先的高性能异构AI优化技术与编译优化技术；
2.针对搜索、推荐、广告、音视频以及大模型场景，优化模型训练和推理场景的性能；
3.与公司各算法部门深度合作，对重点项目进行算法与系统的联合优化。