岗位描述
岗位职责
1. 参与 AI 芯片软件栈总体架构设计,分析软件系统各模块之间的关系、边界和协同方式;
2. 结合 AI 模型、典型算子和硬件架构特征,分析软件栈对端到端性能、资源利用率和开发体验的影响;
3. 识别模型部署和软件执行过程中的关键瓶颈,推动编译器、Runtime、算子库、驱动和工具链等方向的架构优化;
4. 结合 Benchmark、Demo、客户需求和模型部署反馈,持续优化软件栈架构和性能调优方法;
5. 与硬件架构、编译器、算子、驱动、算法、验证和应用团队协同,推动软硬件协同设计和方案落地。
任职要求
1. 熟悉 AI 芯片、NPU、GPU、AI 加速器或异构计算平台的软件栈架构;
2. 理解编译器、Runtime、驱动、算子库、内存管理、任务调度、性能分析等软件系统中的一种或多种方向;
3. 熟悉深度学习模型或典型算子特征,如 CNN、Transformer、Attention、GEMM、Conv、MoE、KV Cache 等;
4. 具备较强的系统分析能力,能够从模型、算子、软件栈和硬件资源之间分析性能瓶颈;
5. 具备良好的架构文档能力,能够输出架构方案、流程设计、接口定义、风险清单和优化建议;
6. 具备跨团队沟通能力,能够与硬件、软件、算法、编译器、算子、驱动和验证团队协同工作。
加分项
1. 有 AI 芯片软件栈、NPU Runtime、AI 编译器、算子库、驱动、SDK 或模型部署相关经验;
2. 有编程模型、编译器、图优化、算子融合、算子适配或性能优化经验;
3. 有大模型推理、Transformer、MoE、KV Cache、Benchmark 性能分析相关经验;
4. 有软硬件协同优化、芯片 Bring-up、SDK 交付、客户模型适配或量产芯片软件栈经验;