GitHub repository: mch5048/G-SWM
Primary language: Python
Description: Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"