扩散模型训练推理国产芯片infra实习生

岗位职责
1、基于LightX2V框架(GitHub - ModelTC/LightX2V: Lightweight Image Video Action Generation Inference Framework)，完成相关的国产芯片的训练推理工具的适配工作，完成多个模型的微调/强化学习训练国产训练适配，参与到一款工业级开源框架的开发中来；
2、构造模型训练推理相关的部分数据；
3、精度/效率评测工作，用于支撑完整的训练推理适配。

岗位要求
1、具备模型训练/推理的基础知识，具备训练/推理相关问题报错的debug能力；
2、熟悉生成类模型（DiT）架构，能够理解模型结构、计算方式和性能瓶颈之间的关系；
3、具有良好的动手能力和分析能力；
4、熟练使用ai工具，用ai辅助写代码的同时，理解你所提交的每一行代码的作用；
5、Base北京，保证线下出勤。