小红书【2027校招】Dots-大模型高性能计算工程师- AI infra
校招全职地点:北京 | 上海状态:招聘♡ 收藏
工作描述
任职要求 1、本科及以上学历,计算机等相关专业方向优先; 2、熟练掌握GPU CUDA编程; 3、熟练掌握Megatron/DeepSpeed/FSDP等框架的研发,熟练掌握模型并行分布式技术; 4、对大语言模型算法感兴趣,有机器学习算法知识背景; 5、追求技术极致,务实,渴望有自己…
登录查看完整工作描述
微信扫码,1秒登录
包括英文材料
学历+
CUDA+
https://developer.nvidia.com/blog/even-easier-introduction-cuda/
This post is a super simple introduction to CUDA, the popular parallel computing platform and programming model from NVIDIA.
https://www.youtube.com/watch?v=86FAWCzIe_4
Lean how to program with Nvidia CUDA and leverage GPUs for high-performance computing and deep learning.
Megatron+
https://www.youtube.com/watch?v=hc0u4avAkuM
DeepSpeed+
[英文] Getting Started
https://www.deepspeed.ai/getting-started/
DeepSpeed model training is accomplished using the DeepSpeed engine.
https://www.youtube.com/watch?v=pDGI668pNg0
FSDP+
https://docs.pytorch.org/tutorials/intermediate/FSDP_tutorial.html
In DistributedDataParallel (DDP) training, each rank owns a model replica and processes a batch of data, finally it uses all-reduce to sync gradients across ranks.
https://www.youtube.com/watch?v=PjEwLgyzuzQ
FSDP provides a comprehensive framework for large model training in PyTorch.
还有更多 •••
相关职位
校招
1、对大语言模型能力边界有清晰认知,知道"什么是好模型"、"好在哪" 2、熟悉主流 benchmark 的设计哲学和局限,不只是会跑分,而是能看出分数的含义 3、有数据处理与工程能力(Python 熟
更新于 2026-10-21北京|上海|杭州

实习VX
1.动画/游戏美术相关专业本科及以上,熟练掌握动画运动规律。 2.熟悉界面动效设计和交互设计的基本原理。 3.热爱游戏 富于创意。有一定的沟通能力。 4.熟练掌握AE(包括粒子系统)、Spine、Fl
更新于 2026-08-13上海

实习TA
1. 2027届毕业生,本科及以上学历,软件工程/计算机/信息技术/人工智能等相关专业在读优先。 2. 对AIGC有浓厚兴趣。 3. 扎实的数学基础,良好的中英文文献阅读能力。 4. 具有良好的编程知
更新于 2026-08-13上海