AI芯片软件实习生
思特威
岗位职责
1、参与面向大语言模型的高性能推理系统研发,在指导下开展核心计算内核、量化、内存访问或多加速设备通信的性能测试、瓶颈分析与优化验证。
2、参与面向大语言模型推理的自研神经网络处理器软件技术预研和原型开发,协助完成编译器、运行时、内核或开发调试工具中的具体模块。
3、参与典型模型适配、精度对齐、性能评测和问题复现,完善基准测试、自动化测试及性能数据整理。
4、协助上层AI应用及智能体项目完成推理环境搭建、模型与服务部署、资源配置和运行问题排查。
5、使用Claude Code、Codex等AI辅助开发工具开展代码理解、技术调研、开发和测试,并对生成代码和实验结果进行验证。
6、整理实验记录、测试结果和技术文档,与团队成员协作完成可验收的研发任务。
任职要求
1、计算机、软件工程、电子信息、微电子、自动化、人工智能或相关专业硕士及以上。
2、熟练掌握C/C++或Python,熟悉Linux开发环境、Git及常用调试工具,具备良好的数据结构与算法基础。
3、了解操作系统、计算机体系结构、存储层次或并行计算等基础知识,具备独立实现、调试和验证代码的能力。
4、了解深度学习基本原理、Transformer及大语言模型推理流程;有模型部署、性能分析或系统软件相关项目经历者优先。
5、接触过以下技术中的一项或多项者优先:PyTorch/ONNX、LLVM/MLIR、OpenAI
Triton、CUDA/ROCm、NCCL/RDMA或大语言模型推理框架。
6、能够使用Claude Code、Codex等AI辅助开发工具进行开发和研究,并能够审查生成代码、补充测试和复现实验结果。
7、对AI芯片和系统软件有强烈兴趣,愿意钻研编译、指令、访存、并行计算和数值精度等技术细节,具备较强的自驱力和动手能力。
8、具备良好的逻辑分析、沟通协作和技术写作能力,能够阅读英文技术资料。