岗位描述(节选)
【岗位职责】
负责大模型分布式训练框架 和 分布式推理框架 研发,主要包括:
1. 基于分布式训练框架(如 Megatron-LM、DeepSpeed)或 高性能推理框架(如 SGLang、vLLM)进行二次开发与优化;
2. 分析并优化训练与推理阶段的性能瓶颈,包括显存占用、通信延迟、计算效率等;
3. 研发高效的训练与推理融合框架(如 RLHF),支持大规模分布式环境;
4. 深入研究 CUDA、通信框架(NCCL、RDMA 等),开发高性能算子及优化分布式通信效率。
【任职要求】
1. 学历背景: 985 硕士及以上学历,计算机相关专业,具备扎实的理论基础和强大的动手能力;
2. 编程能力:
- 熟悉 Python 或 C++/CUDA 编程,熟悉 PyTorch 及其底层实现;
- 熟悉分布式系统开发与调试,有多进程调度与并行算法优化经验者优先;
3. 推理与训练经验:
……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。
本站当前在招 9346 个北京岗位(全城各职能合计),其中 2257 个公开标注了薪资。按这些标注统计:月薪(税前)主流区间 5.5K ~ 15K(25~75 分位),中位约 9K,最低 2K、最高 300K。
在招岗位较多的企业:美团 1313 个阿里巴巴集团(含高德/阿里云) 797 个理想汽车 498 个京东 336 个小米集团 331 个
统计口径:求职者营地实时在招岗位的公开薪资标注,未标注者不计入;区间为 25~75 分位,非薪酬承诺。