Contrastive Learning of Structured World Models
☆396Jun 3, 2020Updated 5 years ago
Alternatives and similar repositories for c-swm
Users that are interested in c-swm are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- Official PyTorch implementation of GENESIS and GENESIS-V2☆109Apr 13, 2022Updated 3 years ago
- Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"☆49Dec 24, 2023Updated 2 years ago
- Official data and code for our paper Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learning☆52Jul 15, 2021Updated 4 years ago
- Structured Object-Aware Physics Prediction for Video Modeling and Planning☆32May 9, 2020Updated 5 years ago
- Code for "Unsupervised State Representation Learning in Atari"☆258Nov 2, 2023Updated 2 years ago
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆59May 16, 2022Updated 3 years ago
- CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning☆599Oct 28, 2020Updated 5 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆284Mar 3, 2026Updated 3 weeks ago
- Dream to Control: Learning Behaviors by Latent Imagination☆585Sep 10, 2021Updated 4 years ago
- Burgess et al. "MONet: Unsupervised Scene Decomposition and Representation"☆89Dec 3, 2022Updated 3 years ago
- Repository for the paper "Planning to Explore via Self-Supervised World Models"☆234Feb 10, 2023Updated 3 years ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Dec 8, 2022Updated 3 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Jun 14, 2018Updated 7 years ago
- Code for "Recurrent Independent Mechanisms"☆120Mar 24, 2022Updated 3 years ago
- Reinforcement Learning in PyTorch☆2,273Jan 4, 2021Updated 5 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Simple and easily configurable grid world environments for reinforcement learning☆2,419Mar 2, 2026Updated 3 weeks ago
- Implementation of Sequential Attend, Infer, Repeat (SQAIR)☆96Apr 9, 2019Updated 6 years ago
- CompILE: Compositional Imitation Learning and Execution (ICML 2019)☆112May 12, 2019Updated 6 years ago
- Repository for the paper "Long-Horizon Visual Planning with Goal-Conditioned Hierarchical Predictors"☆46Nov 22, 2022Updated 3 years ago
- Public Release of Plan2vec Implementation in pyTorch☆57Oct 28, 2022Updated 3 years ago
- Dream to Control: Learning Behaviors by Latent Imagination☆710Jul 14, 2020Updated 5 years ago
- Neural relational inference for interacting systems - pytorch☆768Nov 19, 2019Updated 6 years ago
- [ICML 2019] TensorFlow Code for Self-Supervised Exploration via Disagreement☆130Jun 11, 2019Updated 6 years ago
- Implementation of Sum-Product Attend-Infer-Repeat☆31May 21, 2020Updated 5 years ago
- Code for the paper: Putting An End to End-to-End: Gradient-Isolated Learning of Representations☆291Mar 28, 2023Updated 2 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆71Apr 30, 2023Updated 2 years ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆871Oct 14, 2024Updated last year
- Original tensorflow implementation of SILOT (Spatially Invariant, Label-free Object Tracking).☆13Mar 24, 2023Updated 2 years ago
- Mastering Atari with Discrete World Models☆1,015Jan 21, 2023Updated 3 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Sep 3, 2022Updated 3 years ago
- ☆22Sep 22, 2022Updated 3 years ago
- Collections of robotics environments geared towards benchmarking multi-task and meta reinforcement learning☆1,771Jan 20, 2026Updated 2 months ago
- disentanglement_lib is an open-source library for research on learning disentangled representations.☆1,421May 16, 2021Updated 4 years ago
- ☆80Oct 3, 2023Updated 2 years ago
- S.M.Ali Eslam et.al. Attend, Infer, Repeat: Fast Scene Understanding with Generative Models ICML16☆14Sep 27, 2018Updated 7 years ago
- Implementation of VariBAD: A Very Good Method for Bayes-Adaptive Deep RL via Meta-Learning - Zintgraf et al. (ICLR 2020)☆199Mar 15, 2023Updated 3 years ago
- Learning Latent Dynamics for Planning from Pixels☆1,238Mar 24, 2023Updated 2 years ago