达摩院-芯片性能架构师-计算技术

岗位职责

1. 对当前流行的生成式AI模型，如GPT-3，LLaMA，Qwen，stable diffusion等模型进行应用分析，评估模型在不同硬件架构（CPU/GPU/NPU）上的性能瓶颈，分析内存占用等；
2. 针对应用分析结果设计关键硬件模块的架构，包括数据通路，控制逻辑，计算/访存接口等；
3. 构建高精度的硬件架构的仿真器和性能、面积、功耗模型；
4. 依据仿真器和模型的结果做架构探索，为硬件架构设计提供理论依据。

任职要求

1. 具备计算机系统、芯片微架构（尤其是GPU/NPU）等领域的丰富知识；
2. 有较强的编程能力、脚本和数据总结分析能力；
3. 熟悉芯片性能分析方法，对仿真器（如Gem5、SST等）和NPU性能建模（Timeloop、Maestro等）、基准测试方法熟悉者优先；
4. 熟悉深度学习框架，如pytorch、HuggingFace等，了解流行的大语言模型框架，如vLLM、LightLLM者优先；
5. 良好的英语听说读写能力；
6. 有上进心、团队合作精神和良好的沟通能力。

岗位标签

NEW