Job Description

2027校招-AI计算核算法分析工程师 （北京/上海/成都/西安）

北京、上海、成都、西安

校招

正式

职位描述

岗位职责 1. 开展AI加速器性能架构分析与优化： · 面向AI及通用并行计算场景，围绕benchmark→性能度量→瓶颈定位→优化方案流程，分析计算系统性能； · 聚焦计算图优化、AI workload特征、不规则计算、负载均衡等关键方向。 2. 性能分析与瓶颈定位： · 使用性能分析工具及Python/C++脚本，分析AI加速器执行效率； · 定位计算、访存、并行执行、资源利用率、任务调度等方面的性能瓶颈。 3. 优化策略研究与验证: 针对Transformer、LLM、MoE、稀疏计算等典型workload，分析性能特征； 评估算法、编译优化及架构优化策略，输出数据驱动的优化方案。 4. 软硬件协同优化: · 与架构、编译器、算子、软件栈团队协作，推动性能优化方案落地； · 建立实验环境，沉淀性能分析方法和技术报告。

职位要求

任职要求 1. 硕士及以上学历，计算机体系结构、计算机科学、电子信息、微电子等相关专业； 2. 熟练掌握C/C++、Python，具备并行计算或异构计算经验； 3. 理解GPGPU/NPU架构和执行模型，了解计算、访存、并行效率等性能优化方法； 4. 具备性能分析、benchmark设计和问题定位能力。 加分项： · 有GPU/NPU性能优化或架构分析经验； · 熟悉CUDA、OpenCL、ROCm、Ascend等计算平台； · 熟悉CUDA Kernel优化、Triton、CUTLASS、TVM等技术； · 具备数字电路或处理器微架构基础。

投递