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☆74Apr 7, 2026Updated 4 months ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Clockwork Variational Autoencoder☆50Jun 30, 2021Updated 5 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- ☆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
- 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
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆90Jan 24, 2024Updated 2 years ago
- Attend Infer Repeat (AIR) in PyTorch☆22Jul 25, 2024Updated 2 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Jun 14, 2018Updated 8 years ago
- Dreamer on JAX☆16Jan 19, 2022Updated 4 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
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated last year
- 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"☆36Dec 8, 2022Updated 3 years ago
- FID computation in Jax/Flax.☆29Jul 17, 2024Updated 2 years ago
- ☆11Jun 7, 2023Updated 3 years ago
- ☆649Jul 30, 2026Updated 2 weeks ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Nov 6, 2019Updated 6 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.
- An empirical investigation of deep learning theory☆16Oct 3, 2019Updated 6 years ago
- Official code accompanying the arXiv paper Compressing Multisets with Large Alphabets☆30Sep 22, 2021Updated 4 years ago
- Extending rllab to event-driven multiagent environments☆13Oct 1, 2018Updated 7 years ago
- Variational Reinforcement Learning☆18Jul 25, 2024Updated 2 years ago
- Building blocks for productive research☆72Mar 26, 2026Updated 4 months ago
- Implementation of Data Efficient Reinforcement Learning in Pytorch☆20Aug 6, 2019Updated 7 years ago
- ☆67Jun 25, 2021Updated 5 years ago
- [CoRL 2021] A robotics benchmark for cross-embodiment imitation.☆60Oct 4, 2023Updated 2 years ago
- Scalable metrics logging and analysis☆18Jun 30, 2026Updated last month
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- ☆16Sep 25, 2019Updated 6 years ago
- ☆14Jun 26, 2019Updated 7 years ago
- Vectorization techniques for fast population-based training.☆57Apr 26, 2026Updated 3 months 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 5 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆62Aug 4, 2022Updated 4 years ago
- Fast reinforcement learning research☆65May 25, 2026Updated 2 months ago