arnavkj1995 / VSGView external linksLinks
Learning Robust Dynamics Through Variational Sparse Gating
☆20Oct 19, 2022Updated 3 years ago
Alternatives and similar repositories for VSG
Users that are interested in VSG are comparing it to the libraries listed below
Sorting:
- ☆27Mar 20, 2024Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆163Jun 23, 2023Updated 2 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆12Feb 23, 2024Updated last year
- ☆42May 11, 2022Updated 3 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- General Modules for JAX☆72Sep 12, 2025Updated 5 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- Official implementation of "Approximating Gradients for Differentiable Quality Diversity in Reinforcement Learning"☆22Oct 3, 2022Updated 3 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆34Mar 29, 2023Updated 2 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated 9 months ago
- Deep Hierarchical Planning from Pixels☆114Dec 21, 2022Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- Implementation of Data Efficient Reinforcement Learning in Pytorch☆20Aug 6, 2019Updated 6 years ago
- ☆23Aug 19, 2022Updated 3 years ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- Code for our NeurIPS 2021 paper: Sparsely Changing Latent States for Prediction and Planning in Partially Observable Domains☆22Jan 30, 2023Updated 3 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆104May 17, 2022Updated 3 years ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆36May 19, 2023Updated 2 years ago
- ☆52Jan 20, 2023Updated 3 years ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30May 19, 2022Updated 3 years ago
- Clockwork Variational Autoencoder☆50Jun 30, 2021Updated 4 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Official implementation of the Informed Dreamer algorithm, based on DreamerV3☆19Jan 29, 2026Updated 2 weeks ago
- ☆13May 21, 2023Updated 2 years ago
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 3 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆82Mar 9, 2023Updated 2 years ago
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆28Aug 19, 2025Updated 5 months ago
- Extending rllab to event-driven multiagent environments☆13Oct 1, 2018Updated 7 years ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆32Feb 6, 2023Updated 3 years ago
- Implementation of CURIOUS: Intrinsically Motivated Modular Multi-Goal Reinforcement Learning☆27May 15, 2020Updated 5 years ago
- Simple JAX Graphics Library.☆36Nov 3, 2024Updated last year
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- ☆129Jul 30, 2024Updated last year
- ☆21May 20, 2025Updated 8 months ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Feb 27, 2023Updated 2 years ago
- Minimalistic, hackable PyTorch implementation of SimSiam in ~400 lines. Achieves good performance on ImageNet with ResNet50. Features dis…☆21Nov 25, 2024Updated last year
- Code for the paper "Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learning". Great performance in many environments…☆35Oct 24, 2025Updated 3 months ago