Training a joint Space time model

[Vidyaranya]

What changes are to be made to adapt the divided space time attention to a join space-time attention model?

[nikhilCad]

@Vidyaranya did you wrote it?