大模型训练加速工程师-智能创作（上海/深圳）

关于这个机会

1、优化视觉大模型训练流水线，全面提升训练效率、速度和可扩展性； 2、开发并改进分布式训练策略（如数据并行、模型并行、流水线并行及通信优化），加速模型训练过程； 3、对深度学习模型进行基准测试和性能剖析，精准识别性能瓶颈并优化计算资源的利用率。

任职要求

1、计算机、电子、人工智能等相关专业，本科及以上学历； 2、2年以上大模型训练加速或性能优化经验； 3、具备扎实的软件工程技能，熟练掌握Python和C++；能熟练使用主流深度学习框架，如PyTorch； 4、具备深度学习相关的开发经验，了解基本的模型训练流程； 5、了解常见的生成式AI模型架构，如Transformers或Diffusion models。 加分项： 1、精通分布式训练框架底层原理，如Megatron和Deepspeed； 2、具备分布式训练核心技术的丰富实操经验，能熟练处理数据并行、张量并行、序列并行和流水线并行，并擅长解决通信及显存瓶颈； 3、熟练掌握CUDA编程，具备GPU硬件架构理解与算子级性能调优经验。

内推申请说明

本站为独立内推协助平台。申请会交由管理员核实岗位与内推渠道，不等于已在公司官网投递；薪资、岗位状态和实际招聘流程以官方信息为准。