
Video generation has long forced a hard choice: do you want quality or speed? Bidirectional diffusion models look at every frame at once and produce coherent, visually rich clips, but they are slow and expensive to run. Autoregressive models generate frame by frame, enabling real-time streaming and arbitrarily long videos, but they accumulate errors over time and lose track of what happened earlier in the clip. NVIDIA Research's new Flex-Forcing argues you should not have to choose at all.
One model, every mode
Flex-Forcing is a unified training and inference framework that enables a single video diffusion model to seamlessly operate under both bidirectional and autoregressive generation regimes. The key insight is that these two paradigms are not fundamentally different architectures , they are just two extreme points on a continuous spectrum of how much context each frame is allowed to see.
The core idea is a flexible chunking mechanism jointly defined over the temporal axis and denoising steps. This design allows the model to perform flexible chunking according to different device budgets, perform bidirectional inference across chunks for global structure planning while generating frames autoregressively within each chunk for efficient synthesis, and perform any-order, any-timestep autoregressive generation without the strict causal constraint.
Think of it like this: in fully autoregressive mode, every frame is its own chunk of size one. In fully bidirectional mode, all frames are one giant chunk. Flex-Forcing lets you pick anything in between , and crucially, the same trained model handles all of it.
The two problems it had to solve
Unifying these paradigms in a single model requires solving two non-trivial technical problems.
The first is training the model to handle both causal (left-to-right) and non-causal (full-context) attention simultaneously. Flex-Forcing introduces a unified training and inference framework that enables the model to seamlessly operate under both regimes. During training, the team randomly samples different chunk sizes at each step, exposing the model to the full spectrum of attention patterns from strictly causal to fully bidirectional.
Don't miss what's next in AI
Join 300,000+ engineers and researchers who get the signal, not the noise.
- Full access to in-depth AI research breakdowns
- Be the first to know what's trending before it hits mainstream
- Daily curated papers, repos, and industry moves
