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等技术;
· 具备数字电路或处理器微架构基础。