Learning Robust Dynamics Through Variational Sparse Gating
☆21Oct 19, 2022Updated 4 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:
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 5 years ago
- Code for the paper: Sparsely Changing Latent States for Prediction and Planning in Partially Observable Domains☆12Nov 12, 2021Updated 4 years ago
- ☆39Apr 28, 2026Updated 6 months ago
- Code for our NeurIPS 2021 paper: Sparsely Changing Latent States for Prediction and Planning in Partially Observable Domains☆25Jan 30, 2023Updated 3 years ago
- ☆41May 11, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Fast reinforcement learning research☆65May 25, 2026Updated 5 months 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
- 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
- Evaluating long-term memory of reinforcement learning algorithms☆185Jun 23, 2023Updated 3 years ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 4 years ago
- ☆23Aug 19, 2022Updated 4 years ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆24Nov 8, 2024Updated last year
- Dataset Instructions and Tutorials for Submission to Neurips2022☆10Nov 14, 2023Updated 2 years ago
- General Modules for JAX☆75Apr 7, 2026Updated 7 months 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.
- This repository contains the source codes for the paper: "SPACE: A Simulator for Physical Interactions and Causal Learning in 3D Environm…☆15Oct 11, 2021Updated 5 years ago
- ☆13May 21, 2023Updated 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
- Clockwork Variational Autoencoder☆50Jun 30, 2021Updated 5 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆89Jan 24, 2024Updated 2 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆35Mar 29, 2023Updated 3 years ago
- Deep Hierarchical Planning from Pixels☆125Dec 21, 2022Updated 3 years ago
- A dataloader, but for JAX☆20May 17, 2024Updated 2 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 3 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆87Jul 27, 2022Updated 4 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated last year
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆20Feb 27, 2023Updated 3 years ago
- ☆82Jul 8, 2022Updated 4 years ago
- ☆10Dec 10, 2024Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆82Mar 9, 2023Updated 3 years ago
- ☆14Apr 15, 2025Updated last year
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Code for IROS 2022 paper "Teaching Agents how to Map: Spatial Reasoning for Multi-Object Navigation"☆11Mar 20, 2024Updated 2 years ago
- Official implementation of "Approximating Gradients for Differentiable Quality Diversity in Reinforcement Learning"☆21Oct 3, 2022Updated 4 years ago
- Implementation of Data Efficient Reinforcement Learning in Pytorch☆20Aug 6, 2019Updated 7 years ago
- Code release for "HarmonyDream: Task Harmonization Inside World Models" (ICML 2024), https://arxiv.org/abs/2310.00344☆57Jun 27, 2024Updated 2 years ago
- Offline RL experiments☆15Oct 1, 2022Updated 4 years ago
- Code to accompany the paper 'On Pathologies in KL-Regularized Reinforcement Learning from Expert Demonstrations'☆10Apr 16, 2026Updated 6 months ago
- Pytorch version of Dreamer, which follows the original TF v2 codes.☆143Feb 7, 2022Updated 4 years ago