AI Infra Engineer（预训练框架与性能优化）

岗位描述

面向 VLA 与 World Action Model（视频生成与多模态模型）的预训练需求，参与自研训练框架的开发与演进。通过训练性能分析、分布式执行优化、训练稳定性治理与算法工程协作，提升训练效率与资源利用率，降低训练成本，逐步推动预训练、微调、蒸馏与强化学习共用核心基础能力，支撑不同模型和业务场景的持续扩展。 【职位描述】 1. 训练框架研发：负责预训练框架核心模块的设计、开发与迭代，支持模型、数据、优化器… 1. 扎实的 Python 编程与软件工程能力，熟悉 PyTorch，能够独立完成模块设计、开发、调试与验证； 2. 具有大模型训练框架或分布式训练系统研发经验，熟悉 FSDP、M…