Is your feature request related to a problem? Please describe.

We want to have a loss in the latent space throughout the roll-out steps

Describe the solution you'd like

No response

Describe alternatives you've considered

No response

Additional context

No response

Organisation

No response