Clockwork VAEs in JAX/Flax
☆32Jul 16, 2021Updated 5 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:
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 3 years ago
- Reversible programming in Agda☆13Jun 22, 2023Updated 3 years ago
- General Modules for JAX☆75Apr 7, 2026Updated 6 months ago
- Very deep VAEs in JAX/Flax☆47Jun 16, 2021Updated 5 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 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.
- Clockwork Variational Autoencoder☆50Jun 30, 2021Updated 5 years ago
- ☆22Nov 8, 2021Updated 4 years ago
- 📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)☆25Jun 20, 2021Updated 5 years ago
- MineRL Navigate Video Dataset☆13Mar 24, 2021Updated 5 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆186Jun 23, 2023Updated 3 years ago
- Revisiting Peng's Q(lambda) for Modern Reinforcement Learning☆15Jul 23, 2021Updated 5 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 2 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆90Jan 24, 2024Updated 2 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.
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Jun 14, 2018Updated 8 years ago
- ☆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
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated last year
- ☆67Jun 25, 2021Updated 5 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆88Jul 27, 2022Updated 4 years ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Dec 8, 2022Updated 3 years ago
- FID computation in Jax/Flax.☆29Jul 17, 2024Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆11Jun 7, 2023Updated 3 years ago
- ☆657Updated this week
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Nov 6, 2019Updated 6 years ago
- An empirical investigation of deep learning theory☆16Oct 3, 2019Updated 7 years ago
- Official code accompanying the arXiv paper Compressing Multisets with Large Alphabets☆30Sep 22, 2021Updated 5 years ago
- Variational Reinforcement Learning☆18Jul 25, 2024Updated 2 years ago
- Building blocks for productive research☆73Oct 4, 2026Updated last week
- Implementation of Data Efficient Reinforcement Learning in Pytorch☆20Aug 6, 2019Updated 7 years ago
- Fast and reliable distributed systems in Python☆36Mar 25, 2026Updated 6 months 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.
- [CoRL 2021] A robotics benchmark for cross-embodiment imitation.☆61Oct 4, 2023Updated 3 years ago
- Scalable metrics logging and analysis☆20Jun 30, 2026Updated 3 months ago
- ☆16Sep 25, 2019Updated 7 years ago
- implementation of our self-guided and self-regularized actor-critic algorithm☆29Jan 1, 2023Updated 3 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 6 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆63Aug 4, 2022Updated 4 years ago
- POPGym Library in JAX☆14Apr 15, 2024Updated 2 years ago