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
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
- CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning☆599Oct 28, 2020Updated 5 years ago
- Structured Object-Aware Physics Prediction for Video Modeling and Planning☆32May 9, 2020Updated 5 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
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Jun 14, 2018Updated 7 years ago
- Code for "Unsupervised State Representation Learning in Atari"☆259Nov 2, 2023Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination☆583Sep 10, 2021Updated 4 years ago
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆59May 16, 2022Updated 3 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆281Updated this week
- Burgess et al. "MONet: Unsupervised Scene Decomposition and Representation"☆89Dec 3, 2022Updated 3 years ago
- Code for "Recurrent Independent Mechanisms"☆120Mar 24, 2022Updated 3 years ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Dec 8, 2022Updated 3 years ago
- Repository for the paper "Planning to Explore via Self-Supervised World Models"☆234Feb 10, 2023Updated 3 years ago
- Code for the paper: Putting An End to End-to-End: Gradient-Isolated Learning of Representations☆291Mar 28, 2023Updated 2 years ago
- Public Release of Plan2vec Implementation in pyTorch☆57Oct 28, 2022Updated 3 years ago
- Repository for the paper "Long-Horizon Visual Planning with Goal-Conditioned Hierarchical Predictors"☆46Nov 22, 2022Updated 3 years ago
- [ICML 2019] TensorFlow Code for Self-Supervised Exploration via Disagreement☆129Jun 11, 2019Updated 6 years ago
- Reinforcement Learning in PyTorch☆2,274Jan 4, 2021Updated 5 years ago
- Original tensorflow implementation of SILOT (Spatially Invariant, Label-free Object Tracking).☆13Mar 24, 2023Updated 2 years ago
- Implementation of Sum-Product Attend-Infer-Repeat☆31May 21, 2020Updated 5 years ago
- CompILE: Compositional Imitation Learning and Execution (ICML 2019)☆112May 12, 2019Updated 6 years ago
- Neural relational inference for interacting systems - pytorch☆767Nov 19, 2019Updated 6 years ago
- Dream to Control: Learning Behaviors by Latent Imagination☆710Jul 14, 2020Updated 5 years ago
- Collections of robotics environments geared towards benchmarking multi-task and meta reinforcement learning☆1,749Jan 20, 2026Updated last month
- Mastering Atari with Discrete World Models☆1,009Jan 21, 2023Updated 3 years ago
- disentanglement_lib is an open-source library for research on learning disentangled representations.☆1,420May 16, 2021Updated 4 years ago
- Simple and easily configurable grid world environments for reinforcement learning☆2,405Updated this week
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆869Oct 14, 2024Updated last year
- Implementation of Sequential Attend, Infer, Repeat (SQAIR)☆96Apr 9, 2019Updated 6 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 2 years ago
- Learning Latent Dynamics for Planning from Pixels☆1,234Mar 24, 2023Updated 2 years ago
- RAD: Reinforcement Learning with Augmented Data☆416Mar 29, 2021Updated 4 years ago
- ☆54Oct 28, 2021Updated 4 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Sep 3, 2022Updated 3 years ago
- Visual Foresight: Model-Based Deep Reinforcement Learning for Vision-Based Robotic Control☆142Nov 22, 2022Updated 3 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- DrQ: Data regularized Q☆420Jan 13, 2023Updated 3 years ago