硬件加速训练AI Infra工程师-Data

【岗位职责】 1、负责自研硬件训练框架开发（Torch、Megatron、Dtensor等）； 2、对接业务大模型例如豆包、Seedance等模型在自研硬件上训练任务支持； 3、开发DP、SP、TP、PP等分布式并行方式并且优化； 4、训练业务通信、计算、通算融合算子研发以及性能优化。 【任职要求】 1、1-5年异构硬件、GPU相关领域训练开发经验，熟悉PyTorch、FSDP、Megatron-LM、VeRL分布式训练框架； 2、对数据并行、模型并行、分布式数据并行等常见训练模式有深入理解； 3、参与过硬件加速器训练相关研发工作，包含Torch Compiler、算子开发、编译调度优化等； 4、有主动学习、快速解决问题的能力和自我驱动力。 以下为加分项： 1、有大模型训练百、千卡以上集群训练落地经验； 2、有异构硬件、NPU等训练、推理开发经验，对硬件比较了解； 3、熟悉计算机体系结…… 完整岗位描述与官方投递入口,请登录后在岗位详情页查看。

本站当前在招 209 个西安技术研发岗位，其中 43 个公开标注了薪资。按这些标注统计：月薪（税前）主流区间 9K ~ 20K（25~75 分位），中位约 12K，最低 5K、最高 70K。