AI软件实习生

公司简介：墨芯人工智能（Moffett AI）成立于2018年，总部位于中国深圳，在上海、北京、硅谷设有全球办公室。墨芯拥有全球领先、自主研发的稀疏化算法，是双稀疏算法的发明者，拥有全球专利30余项。墨芯通过颠覆性的稀疏计算技术，为AI计算带来从底层架构到硬件的全面革新，从根本上突破算力增长的瓶颈。

一、工作职责：

自研芯片的全套软件生态建设，对标NVIDIA，包括但不限于：

1. 负责底层基础库和工具开发，如类CUDA Runtime、Debug/Profiler工具等

2. 负责基于运行时库之上的各种库的开发，如集合通信库等

3. 负责开源生态的集成，如PyTorch、vLLM、SGLang等

4. 负责 MaaS 相关软件栈开发

5. 负责模型（尤其是大模型）端到端适配、部署以及性能优化

6. 负责对开发的内容进行测试以及文档等相关工作

二、任职要求：

1. 熟悉Linux平台下的开发，具备独立解决问题能力、团队协作能力、沟通能力

2. 擅长使用 AI 进行开发（如Claude code / Codex / Cursor 且不是用的国产模型）

3. 熟练掌握类CUDA平台（CUDA/寒武纪/华为等）的使用，以及相关的定位问题和分析性能的方法

4. 熟悉PyTorch、vLLM 上做大模型推理优化，深入了解过源码更佳

5. 熟悉大模型的结构和各种推理优化的方式（TP/PP/EP、cache优化）

6. 具备完整的端到端落地的思维和能力，以及面向市场与客户的思维

7. 快速学习新技术的能力

能全勤实习，半年左右