JobConnect

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

投递

Skills

  • C/C++
  • Python
  • Performance Analysis
  • GPU/NPU Architecture
  • Parallel Computing
  • Benchmarking
  • CUDA

Related jobs

北京思朗科技有限责任公司Apply for this job