TransDreamer

[redzhepdx]

I don't know if you have ever considered but TSSM idea doesn't look bad and it can be a good add-on for this library.

Official Paper:
https://arxiv.org/pdf/2202.09481

Implementation:
https://github.com/changchencc/TransDreamer

I don't know why they compared it against the dreamer-v2 only.

[belerico]

That would awesome. Have you considered to open a PR with an initial draft?

[redzhepdx]

Maybe in the near future but at this very moment I am quite busy with my on going work. I will visit your PRs and issues, if there is going to be no PR related to this then I will start working on it.