AI 芯片系统软件与架构实习生

职位详情

你会负责

• 支持 AI accelerator benchmark 设计、评估流程搭建和结果分析。
• 对 LLM inference 等代表性工作负载进行 profiling，识别瓶颈。
• 设计 microbenchmark，研究 memory access、utilization、latency 等行为。
• 参与 simulator 环境、配置管理、regression test、trace parser 和实验框架建设。
• 探索 compiler、runtime、operator mapping、scheduling、memory layout、tiling 和 dataflow 原型。
• 硬件可用后，支持 bring-up、功能验证、性能测试和 simulator-to-silicon 对比。

我们希望你具备

• 计算机、电子工程、自动化、软件工程、集成电路、AI 或相关专业，研究生优先，优秀本科生也欢迎。
• 具备扎实计算机基础，对计算机架构、编译器/系统软件、并行计算、高性能计算、AI 推理系统、模型部署或性能优化至少一个方向有兴趣或经验。
• 具备 C/C++ 或 Python 编程能力。
• 具备分析、实验和技术写作能力。
• 学习主动，能在早期项目的不确定性中快速推进。

加分项

• 学习过计算机架构、并行架构、存储系统或编译器。
• 有 benchmarking、profiling、trace analysis 或 performance modeling 经验。
• 熟悉 PyTorch、CUDA、TVM、MLIR、LLVM、Triton、ONNX Runtime、vLLM 等工具。
• 有 simulator、microbenchmark、runtime system 或 kernel optimization 经验。
• 理解 LLM inference、KV Cache、attention 和 memory hierarchy。
• 有研究项目、系统项目或开源贡献。