研究型实习生-Training Infra工程师当面告知
阿里巴巴(中国)有限公司
2027-02-07 21:45:03
其他27届28届29届实习
岗位详情
地区:
杭州市
专业要求:
人工智能计算机科学与技术软件工程
岗位职责:
负责大模型分布式训练基础设施建设,支撑新模型结构快速接入及大规模训练任务稳定、高效运行。
1. 负责新模型结构在 Megatron 等并行训练框架中的适配,包括模型实现、并行切分、权重转换和训练链路接入;
2. 建设和优化多机多卡训练系统,合理组合 TP、PP、DP、CP、EP、SP 等并行策略;
3. 完善分布式 Checkpoint、自动 Resume、故障检测及容错恢复机制,提升长周期训练任务的稳定性;
4. 支持 Attention、稀疏 Attention、线性 Attention、MoE 等模型结构的高效训练;
5. 定位和解决长上下文及超大规模训练中的显存、通信、负载均衡、数据流水线和算子瓶颈;
6. 建设多卡数值对齐、梯度正确性、Checkpoint 一致性、显存和吞吐回归测试体系;
7. 优化混合精度训练、通信计算重叠、显存管理及关键 CUDA/Triton 算子。
招聘要求:
1. 熟练掌握 Python、PyTorch,熟悉 Megatron-Core、Transformer Engine 等大模型训练框架;
2. 有新模型结构接入分布式训练框架的实践经验,能够独立完成模型适配、权重转换和训练验证;
3. 深入理解 TP、PP、DP、CP、EP、SP 等并行策略的原理、约束及组合方式;
4. 理解 Attention、稀疏 Attention、线性 Attention、MoE 等模型结构及其并行训练特点;
5. 熟悉 PyTorch Autograd、混合精度、梯度累积、分布式通信和分布式 Checkpoint;
6. 能够系统定位 Loss 异常、数值不稳定、训练 Hang、OOM、通信超时和性能退化等问题。
7. 具备多机多卡训练的数值正确性与性能测试能力。
8. 具备良好的系统设计、工程实现和复杂问题排查能力。
【加分项】
1. 具备 CUDA、Triton 或其他高性能算子开发和优化经验;
2. 熟悉 NCCL、RDMA、通信拓扑以及通信计算重叠优化;
3. 有 MoE、超长上下文或千卡级大模型训练经验;
4. 熟悉 FP8/BF16 训练、Loss Scaling、累加误差及数值稳定性优化;
5. 有训练平台、自动容错、可观测性或大规模集群调度系统建设经验。
公司介绍

阿里巴巴(中国)有限公司
阿里巴巴(中国)有限公司,成立于2007-03-26,位于浙江,法定代表人:蒋芳,登记状态:存续,注册资本15460.55039万美元,公司类型:民营/私企
公司信息:
公司名称:阿里巴巴(中国)有限公司
公司类型:互联网/电子商务
公司性质:5000以上
公司地点:
公司规模:--
涉及领域:--
公司网址: