小红书
AI Infra网络通信实习生
AI Infra网络通信实习生
发布于 大约 3 小时前实习/见习
杭州市
无经验要求
实习生
仅现场办公
学历未注明
网络工程
Cuda
Megatron-Lm
Nccl
Vllm
分布式训练
性能优化
AI 估算 · 5k–10k
AI基础设施实习岗，技术要求高，公司为一线互联网，实习薪资处于市场偏上水平。
职位详情
关于这个职位
该职位聚焦于AI大模型训练/推理场景下的高性能网络通信优化，你将参与集合通信库（如NCCL）的深度调优，完善通信可观测能力，并设计与训练框架的联合方案
适合对分布式系统和GPU编程有热情的同学，能够深入了解大规模AI集群的通信机制
最低要求
精通cpp语言编程，熟悉Python编程，具备优秀的系统开发、调试，优化能力和经验 2.具备扎实的计算机理论基础，对数据结构和算法有较强的技术功底
有开发/优化集合通信库（如NCCL、DeepEP、MoonCake）、分布式并行优化相关经验，熟悉NvLink、Infiniband、RoCEv2等高性能通信技术
熟悉Cuda开发，了解GPU工作原理，有kernel开发优化经验
了解主流模型及模型框架，熟悉开源常见训练/推理框架（如vLLM、SGLang、Megatron-LM）
优秀的问题分析和解决能力，勇于挑战
良好的沟通能力和团队合作精神，有比较强的推动能力
工作职责
聚焦模型通信场景的深度可观测能力建设，结合小红书场景，完善集合通信、点对点通信等通信方式与训练/推理框架的联合方案设计，提升训练/推理性能
AI 洞察
优缺点分析
优点
- 接触小红书大规模的AI训练集群，实战经验丰富
- 涉及最前沿的GPU通信技术（NCCL、InfiniBand等），技术含金量高
- 大厂实习经历，未来求职有竞争力
- 技术门槛高，需要扎实的C++和底层系统知识
- 实习期间需要快速上手，工作强度可能较大
- 适合对高性能计算和分布式系统有浓厚兴趣，喜欢钻研底层技术的计算机相关专业学生
缺点 / 挑战
- 通信优化问题复杂，调试和排错难度较高
角色解读
- 从通信优化方向深入，成为AI Infra领域专家
- 向分布式系统工程师或高性能计算工程师方向发展
- 未来可晋升为技术专家或团队负责人
- 负责构建模型通信场景的深度可观测能力，分析NCCL等集合通信库的性能瓶颈
- 参与大模型训练/推理框架与通信库的联合优化，提升分布式训练效率
- 研究NvLink、InfiniBand、RoCEv2等高速网络技术，并编写优化代码
- 精通C++和Python编程，具备系统级开发与调试能力
- 熟悉NCCL、CUDA编程，了解GPU架构与kernel优化
- 理解分布式训练框架（如vLLM、Megatron-LM）和集合通信原理
申请策略
- 在简历中体现对AI基础设施的热情，如开源项目贡献或技术博客
- 了解小红书的AI业务方向，在面试中结合场景回答
- 突出C++/Python的项目经验，特别是有高性能或系统编程相关的项目
- 如果参与过分布式训练或NCCL相关项目，一定要重点描述
- 展示对GPU架构和CUDA编程的理解，或相关优化经历
- 提前学习NCCL源码和集合通信算法，了解Ring/Tree等拓扑
- 动手实践CUDA编程，尝试写简单的kernel
- 熟悉vLLM或Megatron-LM等框架的通信机制
面试指南
- 针对原理题，从基础算法入手，结合具体实现细节
- 针对经验题，用STAR法则说明项目背景、任务、行动和结果
- 针对设计题，先明确需求，再给出分层架构和关键指标
- 请解释NCCL中Ring AllReduce的工作原理
- 如何定位分布式训练中的通信瓶颈？
- 你有过CUDA kernel优化的经验吗？具体优化了什么？
- 介绍一下你在项目中遇到的最棘手的系统问题及解决过程
- 如何设计一个通信可观测系统来监控GPU集群的性能？
职位点评
74
综合评分
AI基础设施前沿技术，成长空间大，薪资福利一般，工作模式未明确
从学习成长、工作节奏、岗位方向和实习待遇综合评估，方便比较实习机会。
更适合这类人
适合追求技术成长、希望深入AI底层基础设施的实习生
表现最好
成长发展
相对薄弱
薪资福利
薪资福利50
成长发展95
工作生活60
使命价值70
薪资福利
50较低
实习岗位薪资相对有限，但公司平台和品牌效应有一定补偿。
薪资信号未披露（AI估算：5K-10K/月）
成长发展
95较高
涉及AI基础设施前沿技术，可积累分布式系统和高性能计算经验，成长性强。
技术前沿前沿/新兴技术
技术栈C++、Python、NCCL、CUDA、vLLM、Megatron-LM
业务类型ambiguous
工作生活
60中等
未明确工作模式与加班情况，互联网公司实习可能较灵活但也可能较忙。
工作模式未明确
办公地点未明确
加班情况未提及（无法判断）
使命价值
70中等
AI技术推动产业创新，具有技术意义，但具体社会价值中性。
行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
小红书 的其他在招职位
相似职位推荐
Watch Jobs