快手 AI Infra 校招 (3)

本岗位面向 NV 和 AMD 异构平台优化方向，涉及 AI 编译器、模型部署与推理优化、算子优化、高性能算子生成、TVM、Triton、图优化、量化等领域。

岗位职责

1. 参与 AI 编译器相关项目的技术方案设计与研发，利用 CPU、GPU、DSP 及 NPU 的并行计算优化、架构设计、稀疏优化和异构调度等高性能优化技术，构建高性能异构 AI 推理引擎与编译器。
2. 针对搜索、推荐、广告、音视频及大模型场景，优化模型训练和推理的计算图执行效率。
3. 与各算法部门深度协作，对重点项目进行算法与系统的联合优化。

任职要求（满足其一即可）

1. 具有 LLVM 编译工具链及优化 Pass 开发经验。
2. 精通 C++ 编程，具备扎实的数据结构与算法功底，熟悉计算机体系结构和 X86 汇编，了解 Python 编程。
3. 熟悉 XLA、MLIR、TVM、Triton、TensorRT 等技术并有相应开发经验者优先。
4. 精通 CPU（ARM/x86）或 GPU（Intel/Nvidia/AMD）平台的高性能计算优化，对计算机体系结构有深入理解，熟悉并行计算优化、访存优化和低比特计算等。
5. 了解深度学习算法基本原理，熟悉神经网络基本架构及算子计算方式，了解至少一种深度学习训练框架（如 PyTorch、TensorFlow）及其模型文件解析。
6. 具备 CPU/GPU 算子 Kernel 开发和性能优化经验，熟悉 CUDA 编程。
7. 具备独立解决问题的能力，能对业务逻辑进行合理抽象和拆分，具有良好的团队合作精神。
8. 了解主流 AIGC 算法模型原理，具有 AIGC 模型加速优化经验者优先。