【2026 实习生】软件工程师实习生 - NCCL当面告知
英伟达半导体科技(上海)有限公司
2026-11-22 23:26:40
其他27届28届实习
岗位详情
地区:
上海
专业要求:
计算机科学与技术
岗位职责:
您的职责:
1. 设计、实现并维护针对深度学习框架(例如用于 TensorFlow/PyTorch 的 NCCL)和 HPC 编程接口(例如用于 MPI/OpenSHMEM 的 UCX)在 GPU 集群上的高度优化的通信运行时。
2. 参与并推动并行编程接口规范(如 MPI/OpenSHMEM)的制定。
3. 设计、实现并维护支持 GPU 之间以及 GPU 与系统其他组件之间交互的系统软件。
4. 创建概念验证原型,以评估和推动编程模型扩展、运行时新设计以及硬件新特性。
岗位要求:
1. 您正在攻读 CE(计算机工程)/ CS(计算机科学)/ EE(电子工程)方向的博士学位,并具备扎实的计算机体系结构、操作系统、通信库和/或 AI/ML(人工智能/机器学习)背景。
2. 优秀的 C/C++ 编程和调试能力。
3. 丰富的 Linux 使用经验。
4. 具备并行编程接口和通信运行时的经验。
5. 具备在多国家、多时区的企业环境中高效工作和沟通的能力及灵活性。
加分项(如何脱颖而出):
1. 深入了解 InfiniBand、RoCE 等高性能网络技术。
2. 具有 HPC(高性能计算)应用程序经验。
3. 具有 PyTorch、TensorFlow、JAX/XLA、vLLM/SGLang 等深度学习框架的使用经验。
4. 具有 AI/DL(人工智能/深度学习)通信模式经验,例如 EP(专家并行)、TP(张量并行)、DP(数据并行)、PP(流水线并行),以及了解如何利用 NCCL 实现这些模式。具有 CUDA 内核优化和性能分析经验。
5. 具有大规模模型训练和生产级推理软件栈的经验。
6. 优秀的团队协作和人际沟通能力,特别是在动态矩阵式管理环境中有效引导和影响他人的能力。
招聘要求:
1. 您正在攻读 CE(计算机工程)/ CS(计算机科学)/ EE(电子工程)方向的博士学位,并具备扎实的计算机体系结构、操作系统、通信库和/或 AI/ML(人工智能/机器学习)背景。
2. 优秀的 C/C++ 编程和调试能力。
3. 丰富的 Linux 使用经验。
4. 具备并行编程接口和通信运行时的经验。
5. 具备在多国家、多时区的企业环境中高效工作和沟通的能力及灵活性。
加分项(如何脱颖而出):
1. 深入了解 InfiniBand、RoCE 等高性能网络技术。
2. 具有 HPC(高性能计算)应用程序经验。
3. 具有 PyTorch、TensorFlow、JAX/XLA、vLLM/SGLang 等深度学习框架的使用经验。
4. 具有 AI/DL(人工智能/深度学习)通信模式经验,例如 EP(专家并行)、TP(张量并行)、DP(数据并行)、PP(流水线并行),以及了解如何利用 NCCL 实现这些模式。具有 CUDA 内核优化和性能分析经验。
5. 具有大规模模型训练和生产级推理软件栈的经验。
6. 优秀的团队协作和人际沟通能力,特别是在动态矩阵式管理环境中有效引导和影响他人的能力。
公司介绍
英
英伟达半导体科技(上海)有限公司
英伟达半导体科技(上海)有限公司,成立于2004-07-13,位于上海,法定代表人:MARK STEVEN HOOSE,登记状态:存续,注册资本900万美元,公司类型:外企独资
公司信息:公司名称:英伟达半导体科技(上海)有限公司公司类型:计算机硬件公司性质:1000-4999人公司地点:公司规模:--涉及领域:--公司网址: