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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆33Apr 28, 2026Updated last month
- 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 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆180Jun 23, 2023Updated 2 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆42May 11, 2022Updated 4 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☆23Jan 30, 2023Updated 3 years ago
- ☆23Aug 19, 2022Updated 3 years ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 3 years ago
- General Modules for JAX☆74Apr 7, 2026Updated 2 months ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆36Mar 29, 2023Updated 3 years ago
- Fast reinforcement learning research☆64May 25, 2026Updated 3 weeks ago
- Code for the paper: Sparsely Changing Latent States for Prediction and Planning in Partially Observable Domains☆11Nov 12, 2021Updated 4 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆88Jul 27, 2022Updated 3 years ago
- Deep Hierarchical Planning from Pixels☆120Dec 21, 2022Updated 3 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- ☆14Apr 15, 2025Updated last year
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated last year
- Clockwork Variational Autoencoder☆50Jun 30, 2021Updated 4 years ago
- Official implementation of "Approximating Gradients for Differentiable Quality Diversity in Reinforcement Learning"☆22Oct 3, 2022Updated 3 years ago
- ☆10Dec 10, 2024Updated last year
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆82Jun 4, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆10Oct 1, 2018Updated 7 years ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆39May 19, 2023Updated 3 years ago
- Official implementation of the ICML 2025 paper "SOLD: Slot Object-Centric Latent Dynamics Models for Relational Manipulation Learning fro…☆20Sep 30, 2025Updated 8 months ago
- Official Implementation of SFM and the baselines in Jax.☆21May 31, 2025Updated last year
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- Extending rllab to event-driven multiagent environments☆13Oct 1, 2018Updated 7 years ago
- Implementation of Data Efficient Reinforcement Learning in Pytorch☆20Aug 6, 2019Updated 6 years ago
- A topic model which can identify bilingual topics across unaligned corpus using dictionary. An implementation of the paper "Detecting Com…☆14Oct 25, 2017Updated 8 years ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆33Feb 6, 2023Updated 3 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆29Jan 14, 2025Updated last year
- Official implementation of the Informed Dreamer algorithm, based on DreamerV3☆22Jan 29, 2026Updated 4 months ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- ☆13May 21, 2023Updated 3 years ago
- Simple JAX Graphics Library.☆38Nov 3, 2024Updated last year
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Feb 27, 2023Updated 3 years ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆32Oct 12, 2023Updated 2 years ago