岗位描述(节选)
【岗位职责】
1、优化视觉大模型训练流水线,全面提升训练效率、速度和可扩展性;
2、开发并改进分布式训练策略(如数据并行、模型并行、流水线并行及通信优化),加速模型训练过程;
3、对深度学习模型进行基准测试和性能剖析,精准识别性能瓶颈并优化计算资源的利用率。
【任职要求】
1、计算机、电子、人工智能等相关专业,本科及以上学历;
2、2年以上大模型训练加速或性能优化经验;
3、具备扎实的软件工程技能,熟练掌握Python和C++;能熟练使用主流深度学习框架,如PyTorch;
4、具备深度学习相关的开发经验,了解基本的模型训练流程;
5、了解常见的生成式AI模型架构,如Transformers或Diffusion models。
加分项:
1、精通分布式训练框架底层原理,如Megatron和Deepspeed;
2、具备分布式训练核心技术的丰富实操经验,能熟练处理数据并行、张量并行、序列……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。
本站当前在招 2772 个上海技术研发岗位,其中 302 个公开标注了薪资。按这些标注统计:月薪(税前)主流区间 15K ~ 26K(25~75 分位),中位约 20K,最低 4K、最高 100K。
在招岗位较多的企业:字节跳动 361 个拼多多 179 个米哈游 161 个哈啰出行 141 个阿里巴巴集团(含高德/阿里云) 110 个
统计口径:求职者营地实时在招岗位的公开薪资标注,未标注者不计入;区间为 25~75 分位,非薪酬承诺。