岗位描述
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等方面的知识或经验； 职位类别: 社会招聘