arnavkj1995 / VSG
Learning Robust Dynamics Through Variational Sparse Gating
☆21Updated 2 years ago
Alternatives and similar repositories for VSG:
Users that are interested in VSG are comparing it to the libraries listed below
- Sandbox environment for generalizable agent research☆24Updated 2 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 2 years ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Updated 2 years ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- ☆23Updated 2 years ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆15Updated 2 years ago
- Code for "Possibility Before Utility: Learning And Using Hierarchical Affordances" (ICLR 2022)☆14Updated 3 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- ☆44Updated last year
- ☆47Updated 2 years ago
- Standalone library of frequently-used wrappers for dm_env environments.☆17Updated 8 months ago
- Fast reinforcement learning research☆58Updated 3 months ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆32Updated 2 years ago
- ☆24Updated 9 months ago
- Change-Based Exploration Transfer☆36Updated 2 years ago
- ☆42Updated 4 years ago
- General Modules for JAX☆64Updated last month
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆85Updated last year
- PlaNet: Learning Latent Dynamics for Planning from Pixels☆10Updated 5 years ago
- Reinforcement Learning via Supervised Learning☆71Updated 2 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Clockwork Variational Autoencoder☆47Updated 3 years ago
- Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)☆20Updated 3 years ago
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆22Updated last year
- Generalised UDRL☆37Updated 2 years ago
- Estimating Q(s,s') with Deep Deterministic Dynamics Gradients☆32Updated 5 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 7 months ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆80Updated 2 years ago