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

【岗位职责】 面向 VLA 与 World Action Model（视频生成与多模态模型）的预训练需求，参与自研训练框架的开发与演进。通过训练性能分析、分布式执行优化、训练稳定性治理与算法工程协作，提升训练效率与资源利用率，降低训练成本，逐步推动预训练、微调、蒸馏与强化学习共用核心基础能力，支撑不同模型和业务场景的持续扩展。 【职位描述】 1. 训练框架研发：负责预训练框架核心模块的设计、开发与迭代，支持模型、数据、优化器与训练策略的灵活接入；建设统一的训练执行与状态管理机制，完善 checkpoint 保存恢复、指标记录与评测接口，保障训练正确性、可复现性与兼容性； 2. 端到端性能优化：建立性能分析与评测体系，定位数据处理、计算、显存和通信等环节的瓶颈；优化数据加载与预处理流水线、模型计算与显存使用，评估算子优化、编译、低精度等技术的收益，提升 GPU 有效利用率与训练吞吐； 3.…… 完整岗位描述与官方投递入口,请登录后在岗位详情页查看。