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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- MuJoCo models for Unitree Robots☆12Nov 24, 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
- ☆22Nov 8, 2021Updated 4 years ago
- 📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)☆25Jun 20, 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.
- General Modules for JAX☆73Apr 7, 2026Updated last month
- Revisiting Peng's Q(lambda) for Modern Reinforcement Learning☆15Jul 23, 2021Updated 4 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
- Attend Infer Repeat (AIR) in PyTorch☆22Jul 25, 2024Updated last year
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Jun 14, 2018Updated 7 years ago
- MineRL Navigate Video Dataset☆13Mar 24, 2021Updated 5 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated last year
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆86Jul 27, 2022Updated 3 years ago
- Scalable metrics logging and analysis☆18Apr 9, 2025Updated last year
- 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.
- Fast and reliable distributed systems in Python☆35Mar 25, 2026Updated 2 months ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆36Dec 8, 2022Updated 3 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆89Jan 24, 2024Updated 2 years ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Nov 6, 2019Updated 6 years ago
- Variational Reinforcement Learning☆17Jul 25, 2024Updated last year
- Building blocks for productive research☆71Mar 26, 2026Updated 2 months ago
- ☆67Jun 25, 2021Updated 4 years ago
- [CoRL 2021] A robotics benchmark for cross-embodiment imitation.☆60Oct 4, 2023Updated 2 years ago
- Implementation of Data Efficient Reinforcement Learning in Pytorch☆20Aug 6, 2019Updated 6 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.
- ☆14Jun 26, 2019Updated 6 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- A PyTorch re-implementation of "Doubly Reparameterized Gradient Estimators for Monte Carlo Objectives"☆18Sep 11, 2019Updated 6 years ago
- End-to-end training of Retrieval-Augmented LMs (REALM, RAG)☆23Nov 22, 2023Updated 2 years ago
- Accompanying repository for Unsupervised Active Domain Randomization in Goal-Directed RL☆12Aug 4, 2020Updated 5 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆176Jun 23, 2023Updated 2 years ago
- Estimating Q(s,s') with Deep Deterministic Dynamics Gradients☆32Feb 21, 2020Updated 6 years ago
- Stochastic Latent Actor-Critic: Deep Reinforcement Learning with a Latent Variable Model☆154Oct 26, 2020Updated 5 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Repository for the paper "Planning to Explore via Self-Supervised World Models"☆238Feb 10, 2023Updated 3 years ago
- ☆16Jun 25, 2022Updated 3 years ago
- Repo for our ICML21 paper, Unsupervised Learning of Visual 3D Keypoints for Control☆46Feb 17, 2025Updated last year
- Accelerated Methods for Deep Reinforcement Learning☆49Mar 20, 2019Updated 7 years ago
- Learning Action-Value Gradients in Model-based Policy Optimization☆32Sep 7, 2021Updated 4 years ago
- A custom open ai gym environment for solo experimentation.☆12Apr 14, 2021Updated 5 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