墨芯人工智能科技(深圳)有限公司
AI软件开发工程师-上海
RESPONSIBILITIES
岗位职责
自研芯片的全套软件生态建设,对标NVIDIA,包括但不限于:
负责底层基础库和工具开发,如类CUDA Runtime、Debug/Profiler工具等
负责基于运行时库之上的各种库的开发,如集合通信库等
负责开源生态的集成,如PyTorch、vLLM、SGLang等
负责 MaaS 相关软件栈开发
负责模型(尤其是大模型)端到端适配、部署以及性能优化
负责对开发的内容进行测试以及文档等相关工作
REQUIREMENTS
任职要求
熟悉Linux平台下的开发,具备独立解决问题能力、团队协作能力、沟通能力
必须擅长使用 AI 进行开发(如Claude code / Codex / Cursor 且不是用的国产模型)
掌握类CUDA平台(CUDA/寒武纪/华为等)的使用,以及相关的定位问题和分析性能的方法的优先
熟悉PyTorch、vLLM 上做大模型推理优化,深入了解过源码更佳
熟悉大模型的结构和各种推理优化的方式(TP/PP/EP、cache优化)
具备完整的端到端落地的思维和能力,以及面向市场与客户的思维
快速学习新技术的能力
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。
