AI架构师& AI架构性能建模工程师(J10673)

职位描述

AI架构师（偏AI硬件架构设计）&AI架构性能建模工程师（偏AI算法、workload性能分析）

职责

- 对接产品经理、客户支持工程师、算法工程师、框架工程师、软件栈工程师，深入理解AI客户底层需求、并下沉出架构优化方向及具体的设计需求；

- 分析AI/HPC等计算workload，进行GPU、NPU性能建模与分析；

- 针对AI workload作GPU架构增强、设计NPU架构等；

任职要求

- 经验要求

- 初级：硕士+2年以上工作经验

- 中级：硕士+5年以上工作经验，或，博士+2年以上工作经验

- 高级：硕士+8年以上工作经验，或，博士+5年以上工作经验

- 技能要求：

- 理解AI算法及其在GPU上的运行flow；

- 理解以Transformer模型为基础的LLM、以deepseek为代表的现代MoE模型的架构，及其在训练、推理集群上的运行原理；

- 理解算例带宽比的配平原理、roofline model等；

- 能针对大量性能数据进行综合分析、识别计算瓶颈、提出架构建议；

- 架构师：具有编程模型、memory model、软硬接口、ISA的知识和经验；

- 性能建模工程师：能够分析AI模型源代码，并针对目标硬件架构作高层次性能建模；

- 加分项：能分析GPU kernel、有算子设计或分析经验，能进行性能问题分析；

- C++或PyThon编程能力；

- 加分项：NVSHMEM、NCCL、MPI、PGAS等方面的知识或经验；

- 加分项：PTX、nvcc、CUDA C++、CUTLASS、cuTile、Triton、cuBLUS、cuDNN、TensorRT、PyTorch等方面的知识或经验；