2026-09-29
Trainium内核优化路径:实现实时视频生成
Reactor与亚马逊Neuron Science团队合作,针对自回归扩散视频模型Rolling Forcing进行优化,通过NKI自定义内核、混合分片策略及模型结构调整,将3D-RoPE计算从5秒降至1.8毫秒,成功在...
Reactor与亚马逊Neuron Science团队合作,针对自回归扩散视频模型Rolling Forcing进行优化,通过NKI自定义内核、混合分片策略及模型结构调整,将3D-RoPE计算从5秒降至1.8毫秒,成功在...