岗位说明
1. 围绕大模型异构训练集群,尤其是非NV GPU大规模异构硬件集群,面向大模型预训练、Finetune等场景进行新模型使能,系统分析/优化,业务支撑等工作,包括分布式优化框架,AI框架,网络集合通信,算子等方面内容
2. 参与美团机器学习平台建设,构建非NV GPU大规模训练场景软件栈和算法迭代Pipeline
3. 负责NLP类大模型的训练端到端流程,包括训练使能,系统调优,关键指标分析和统计,问题定位等工作,支持常见生成式NLP类大模型,包括但不限于LlaMa 1/2, GPT, BLOOM等
4. 负责视觉类,多模态大模型的训练端到端流程,包括模型使能,系统调优,关键指标分析和统计,问题定位等工作,支持常见的视觉类,多模态大模型,包括但不限于ViT,Swin Transformer,Stable Diffusion, MoE等
5. 负责大模型分布式优化框架的系统分析,性能调优,特性开发,问题定位等工作,支持常见的大模型分布式优化框架,包括但不限于Megatron, DeepSpeed, FSDP等
6. 负责PyTorch在非NV GPU场景下系统分析,性能调优,问题定位等工作
7. 负责常见的大模型训练场景下高性能Kernel算子的关键指标分析和统计,融合开发,性能优化等工作
8. 负责大模型不同规模卡数训练场景下集合通信/NCCL关键指标分析和统计,系统定位/调优等工作
9. 负责与NV GPU相同场景进行Benchmark评测,对比验证,性能分析等工作
岗位要求
业界前列的NV GPU和非GPU算力规模,协同算法,AI框架,网络,计算,芯片等多个团队共同建设大模型软、硬件技术底座,接触最前沿的AI Infra基础设施,结合最前沿的AI平台软件栈,支撑最前沿的大模型不同领域的算法探索,从算法到算子,从芯片架构到互联集群,从POC到大规模场景,充满机遇与挑战,兼顾学习和成长,在垂直,水平等多个领域和维度,全面感受大模型带来的技术变革!
美团是北京三快在线科技有限公司旗下品牌。美团是一家科技零售公司
美团以“零售 + 科技”的战略践行“帮大家吃得更好,生活更好”的公司使命。
自2010年3月成立以来,美团持续推动服务零售和商品零售在需求侧和供给侧的数字化升级,和广大合作伙伴一起努力为消费者提供品质服务。2018年9月20日,美团在港交所挂牌上市。
美团始终以客户为中心,不断加大在新技术上的研发投入。美团会和大家一起努力,更好承担社会责任,更多创造社会价值。
公司自成立以来始终以客户为中心,不断提升服务理念和标准,以便更好承担社会责任,创造更多社会价值。