Cross-Attention Encoder
observations
actions (posterior) / masks (prior)
latents (autoregressive)
bos
z
1
z
1
z
2
z
2
z
3
z
3
eos
Cross-Attention Decoder