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
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 2 years ago
- ☆23Updated 2 years ago
- Sandbox environment for generalizable agent research☆24Updated 2 years ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Updated last year
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Code for "Possibility Before Utility: Learning And Using Hierarchical Affordances" ICLR 2022☆14Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated 2 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆85Updated last year
- Change-Based Exploration Transfer☆36Updated 2 years ago
- Reward shaping approach for instruction following settings, leveraging language at multiple levels of abstraction.☆19Updated 3 years ago
- General Modules for JAX☆62Updated 6 months 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
- Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)☆34Updated last year
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆25Updated last year
- Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)☆20Updated 3 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆53Updated 2 years ago
- ☆24Updated 7 months ago
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆31Updated 3 years ago
- Fast reinforcement learning research☆56Updated last month
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆32Updated last year
- ☆46Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated last year
- ☆35Updated 2 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Updated 3 years ago
- ☆41Updated last year
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆22Updated last year
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- Bingham Policy Parameterization for 3D Rotations in Reinforcement Learning☆24Updated 2 years ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆60Updated 7 months ago
- Standalone library of frequently-used wrappers for dm_env environments.☆18Updated 6 months ago