岗位描述(节选)
作为阿里巴巴视觉生成大模型Al Infra训推优化团队,我们支持了通义Wan和Happy系列模型,你将有机会参与或负责以下工作内容:
● 设计和实现高效的分布式训练架构,优化训练过程中的计算、通信与显存使用,提升 GPU 利用率与整体训练吞吐;
● 探索并实现低bit量化技术和稀疏化,减少模型存储和计算资源消耗,同时保持训练精度;
● 构建训练性能分析与调优工具链,持续定位和解决系统瓶颈;
● 持续关注并跟进业界技术发展,积极尝试和探索新的训练优化方向,提出并验证创新性的解决方案。
【任职要求】
● 计算机、软件工程、人工智能相关专业硕士及以上学历;
● 扎实的工程能力,优良的编程风格,熟悉Python/C++语言和常用设计模式,具备复杂系统的设计开发调试能力;
● 熟悉深度学习基础理论,理解主流训练框架(如 PyTorch、Megatron、DeepSpeed、JAX 等)的特点与实现……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。