AoiZora：面向视频扩散 Transformer 的编译器无关自动并行化框架

AoiZora 是一款面向视频扩散 Transformer 的编译器无关自动并行化框架。该框架采用双 IR 架构，能在 TPU v5e 上将 Wan 2.1 模型的推理去噪延迟降低 1.23 倍。 在性能提升方面，该框架支持 TPU v5p 架构，可将模型训练加速 1.11 至 1.21 倍。其核心机制允许对 GPU 上的 PyTorch 程序进行并行化，并且具备快速求解能力，能在 10 分钟内找到优化方案。 该框架的最大技术特点在于，它无需执行候选方案即可对编译程序进行定价，从而实现了高效的自动并行化流程。这一特性使其在 TPU 和 GPU 环境下均能高效运作。

AI 根据报道生成 · 1 小时前更新

还没有足够的连续观测数据，暂不绘制趋势。