Pinned Loading

1. forgetting-transformer forgetting-transformer Public

   [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"

   Python 84 1

2. SPACE SPACE Public

   Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"

   Python 102 29

3. G-SWM G-SWM Public

   Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"

   Python 35 6