岗位职责
- 负责视频生成/图像生成/多模态等方向模型后训练加速,如SFT、RL、CT、LoRA任务的训练加速;
- 负责Seedance/Seedream模型训练的异构卡适配工作。
职位要求
- 有AI工程系统优化经验,深入理解计算机体系结构、熟悉视觉大模型的原理与分布式训练方法;
- 了解主流AIGC、多模态算法模型原理,有AIGC模型训练适配和加速经验优先;
- 熟悉PyTorch、Megatron、DeepSpeed等训练框架原理,与ZeRO、TP、SP等并行技术原理优先;
- 熟悉Gradient Checkpoint、Activation Offloading、通算融合、同算掩盖、低精度训练等加速技巧优先;
- 熟悉Triton、TileLang、Cute DSL等语言并具备相关技术优化经验,或熟悉Verl、OpenRLHF、TRL优先。