All jobsIT & SoftwareData Analysis and MathematicsAI加速器计算前端算法分析工程师
AI加速器计算前端算法分析工程师
- 北京思朗科技有限责任公司
- Shanghai, Beijing +4 more, China
- CNY 250,000 – CNY 400,000
北京、上海、成都、西安校招正式
职位描述
岗位职责
- 开展AI加速器性能架构分析与优化:
· 面向AI及通用并行计算场景,围绕benchmark→性能度量→瓶颈定位→优化方案流程,分析计算系统性能;
· 聚焦计算图优化、AI workload特征、不规则计算、负载均衡等关键方向。
- 性能分析与瓶颈定位:
· 使用性能分析工具及Python/C++脚本,分析AI加速器执行效率;
· 定位计算、访存、并行执行、资源利用率、任务调度等方面的性能瓶颈。
- 优化策略研究与验证:
针对Transformer、LLM、MoE、稀疏计算等典型workload,分析性能特征;
评估算法、编译优化及架构优化策略,输出数据驱动的优化方案。
- 软硬件协同优化:
· 与架构、编译器、算子、软件栈团队协作,推动性能优化方案落地;
· 建立实验环境,沉淀性能分析方法和技术报告。
职位要求
任职要求
- 硕士及以上学历,计算机体系结构、计算机科学、电子信息、微电子等相关专业;
- 熟练掌握C/C++、Python,具备并行计算或异构计算经验;
- 理解GPGPU/NPU架构和执行模型,了解计算、访存、并行效率等性能优化方法;
- 具备性能分析、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
AI仿真与验证平台软件开发工程师- NIO
- Shanghai, Hefei, China
- CNY 200,000 – CNY 300,000New
端侧AI芯片推理基座软件开发工程师- NIO
- Shanghai, Hefei, China
- CNY 300,000 – CNY 450,000New
数字化工程师- NIO
- Hefei, China
- CNY 200,000 – CNY 300,000New
安全研究员- 亚信安全
- Beijing, Chengdu +2 more, China
- CNY 200,000 – CNY 350,000New
软硬融合-性能&内存优化工程师- Xiaomi
- Shanghai, Xi’an, China
- CNY 400,000 – CNY 800,000New
Equipment Engineer- Western Digital
- Shenzhen, China
- CNY 250,000 – CNY 400,000New
AI Infrastructure Intern- Bambu Lab
- Shanghai, Shenzhen, China
- CNY 200,000 – CNY 300,000
IC验证工程师- Xiaomi
- Shanghai, Beijing +2 more, China
芯片验证工程师- Xiaomi
- Shanghai, Shenzhen +1 more, China
SOC软件测试开发工程师- Xiaomi
- Shanghai, Beijing +1 more, China
- CNY 200,000 – CNY 350,000