岗位职责
作为阿里巴巴视觉生成大模型Al Infra训推优化团队，我们支持了通义Wan和Happy系列模型，你将有机会参与或负责以下工作内容： ● 设计和实现高效的分布式训练架构，优化训练过程中的计算、通信与显存使用，提升 GPU 利用率与整体训练吞吐； ● 探索并实现低bit量化技术和稀疏化，减少模型存储和计算资源消耗，同时保持训练精度； ● 构建训练性能分析与调优工具链，持续定位和解决系统瓶颈； ● 持续关注并跟进业界技术发展，积极尝试和探索新的训练优化方向，提出并验证创新性的解决方案。
职位要求
● 计算机、软件工程、人工智能相关专业硕士及以上学历； ● 扎实的工程能力，优良的编程风格，熟悉Python/C++语言和常用设计模式，具备复杂系统的设计开发调试能力； ● 熟悉深度学习基础理论，理解主流训练框架（如 PyTorch、Megatron、DeepSpeed、JAX 等）的特点与实现细节； ● 熟悉计算机体系结构，具备高性能 Kernel 开发（CUDA / Triton）、分布式策略开发等方面的实践经验； ● 良好的沟通能力与团队协作意识，具备快速学习和深入钻研复杂技术问题的能力。 加分项： ● 具备文生视频 / 图生视频等方向的训练或推理优化实战经验； ● 熟悉Diffusion等视频生成体系，并参与过性能优化或工程落地； ● 对视频生成的系统瓶颈、成本模型和工程化落地有深入思考。
