高性能计算实习生
岗位摘要
负责大规模深度学习训练/推理框架中 CUDA 内核的性能优化与定制开发；开展与 GPU 体系结构相关的前沿探索性工作，包括新指令集、算子融合、异构计算和硬件趋势研究；与算法研究员合作，推动模型结构与底层实现的协同优化
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口，并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 负责大规模深度学习训练/推理框架中 CUDA 内核的性能优化与定制开发
- 开展与 GPU 体系结构相关的前沿探索性工作，包括新指令集、算子融合、异构计算和硬件趋势研究
- 与算法研究员合作，推动模型结构与底层实现的协同优化
任职要求
- 计算机、电子工程或相关专业本科及以上学历
- 精通 CUDA 编程模型，熟悉 Tensor Core、WGMMA、Shared Memory 等底层细节
- 具备优秀的代码实现能力（C++/CUDA/Triton/CUTLASS），良好的工程化与调试能力
- 具备良好的沟通与协作能力，能够与算法研究员、硬件工程师及跨团队成员高效配合
- 熟悉并行算法设计与优化，有大规模矩阵乘 (GEMM)、注意力 (FlashAttention/MLA)、MoE、稀疏算子优化经验者优先
- 熟悉分布式训练/推理框架（Megatron、vLLM、SGLang、TensorRT-LLM 等）者优先
- 参与过开源社区（NCCL、CUTLASS、Triton、TileLang、Megatron-LM、vLLM 等）贡献者优先
福利待遇
- 提供日薪500-1000元的丰厚报酬
- 加入月之暗面顶尖AI技术团队，参与前沿大模型研发
- 深入GPU架构与底层计算的核心技术领域
工作地址
北京海淀区京东科技大厦13层
登录后查看完整岗位详情
使用邮箱验证码登录后，可查看完整职责、任职要求、原岗位入口，并继续使用收藏和浏览记录。列表摘要保持公开，完整详情用于保护数据质量和降低恶意抓取。