Clockwork VAEs in JAX/Flax
☆32Jul 16, 2021Updated 4 years ago
Alternatives and similar repositories for cwvae-jax
Users that are interested in cwvae-jax are comparing it to the libraries listed below
Sorting:
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- 📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)☆25Jun 20, 2021Updated 4 years ago
- ☆22Nov 8, 2021Updated 4 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- Clockwork Variational Autoencoder☆50Jun 30, 2021Updated 4 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Jun 14, 2018Updated 7 years ago
- 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
- Revisiting Peng's Q(lambda) for Modern Reinforcement Learning☆15Jul 23, 2021Updated 4 years ago
- General Modules for JAX☆72Feb 21, 2026Updated 2 weeks ago
- MineRL Navigate Video Dataset☆13Mar 24, 2021Updated 4 years ago
- Variational Reinforcement Learning☆17Jul 25, 2024Updated last year
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Jul 6, 2023Updated 2 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆88Jan 24, 2024Updated 2 years ago
- [CoRL 2021] A robotics benchmark for cross-embodiment imitation.☆60Oct 4, 2023Updated 2 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated 10 months ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- Attend Infer Repeat (AIR) in PyTorch☆22Jul 25, 2024Updated last year
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Dec 8, 2022Updated 3 years ago
- Implementation of Data Efficient Reinforcement Learning in Pytorch☆20Aug 6, 2019Updated 6 years ago
- ☆28Jan 11, 2021Updated 5 years ago
- A custom open ai gym environment for solo experimentation.☆12Apr 14, 2021Updated 4 years ago
- FID computation in Jax/Flax.☆29Jul 17, 2024Updated last year
- Accompanying repository for Unsupervised Active Domain Randomization in Goal-Directed RL☆12Aug 4, 2020Updated 5 years ago
- Recyclable Gaussian Processes☆11Dec 8, 2022Updated 3 years ago
- A set of environments utilizing pybullet for simulation of robotic manipulation tasks.☆29Mar 8, 2021Updated 4 years ago
- Fast and reliable distributed systems in Python☆33Jan 12, 2026Updated last month
- Repository for the paper "Long-Horizon Visual Planning with Goal-Conditioned Hierarchical Predictors"☆46Nov 22, 2022Updated 3 years ago
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Jun 30, 2020Updated 5 years ago
- ☆14Jun 26, 2019Updated 6 years ago
- PyTorch implementation of Count-Based Exploration with Neural Density Models☆10Mar 22, 2018Updated 7 years ago
- Scalable metrics logging and analysis☆18Apr 9, 2025Updated 10 months ago
- Stochastic Latent Actor-Critic: Deep Reinforcement Learning with a Latent Variable Model☆154Oct 26, 2020Updated 5 years ago
- The Variational Homoencoder: Learning to learn high capacity generative models from few examples☆34Jul 13, 2023Updated 2 years ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Nov 6, 2019Updated 6 years ago
- Building blocks for productive research☆69Jan 12, 2026Updated last month
- Learning Action-Value Gradients in Model-based Policy Optimization☆32Sep 7, 2021Updated 4 years ago
- Proto-RL: Reinforcement Learning with Prototypical Representations☆86Jun 12, 2022Updated 3 years ago
- This packages provides a simple python implementation of Invariant Causal Prediction (ICP)☆13Mar 22, 2024Updated last year