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
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆85Updated last year
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 2 years ago
- Sandbox environment for generalizable agent research☆25Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆37Updated 2 years ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- ☆23Updated 2 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆32Updated 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
- General Modules for JAX☆64Updated 2 weeks ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆26Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆17Updated 9 months ago
- ☆44Updated last year
- Change-Based Exploration Transfer☆36Updated 3 years ago
- ☆20Updated 2 years ago
- Fast reinforcement learning research☆59Updated 4 months ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆54Updated 2 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Updated 2 years ago
- ☆42Updated 2 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Generalised UDRL☆37Updated 2 years ago
- ☆22Updated 2 years ago
- Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)☆20Updated 3 years ago
- ☆42Updated 4 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 8 months ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Updated 2 years ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30Updated 2 years ago
- Bingham Policy Parameterization for 3D Rotations in Reinforcement Learning☆24Updated 3 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Updated 3 years ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆42Updated last week