Popular repositories Loading

1. drstrategy drstrategy Public

   This is the official code for the paper "Dr. Strategy: Model-Based Generalist Agents with Strategic Dreaming" (ICML 2024).

   Python 8

2. dreamweaver-release dreamweaver-release Public

   Official Code for "Dreamweaver: Learning Compositional World Models from Pixels"

   Python 4

3. OCVT OCVT Public

   Official PyTorch implementation of "Generative Video Transformer: Can Objects be the Words?"

   2

4. gpu_testing gpu_testing Public

   To find the failing GPUs in a server.

   Python 2

5. SPACE SPACE Public

   Forked from zhixuan-lin/SPACE

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

   Python 1

6. G-SWM G-SWM Public

   Forked from zhixuan-lin/G-SWM

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

   Python