Neuron Science 与 reactorworld 为实时扩散视频构建 Trainium 路径
Amazon 的 Neuron Science 团队与 reactorworld 使用 Neuron Kernel Interface,在 Trainium 上构建了通往实时自回归扩散视频生成的以内核为中心的路径。
Amazon Science 表示,这项工作针对动态形状、内存访问模式和缓存管理;这些问题使此类工作负载难以由通用编译器处理,相关技术可跨模型泛化。该方法见在 Trainium 上实现实时视频生成的以内核为中心的路径。