JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
☆42Jun 6, 2024Updated last year
Alternatives and similar repositories for jax-vqvae-vqgan
Users that are interested in jax-vqvae-vqgan are comparing it to the libraries listed below
Sorting:
- ☆18Apr 28, 2024Updated last year
- Code for minimum-entropy coupling.☆32Jan 6, 2026Updated 2 months ago
- Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"☆21Oct 6, 2021Updated 4 years ago
- Official PyTorch Implementation for Metric Residual Networks for Sample Efficient Goal-Conditioned Reinforcement Learning☆19Jan 11, 2023Updated 3 years ago
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆22Jul 21, 2025Updated 7 months ago
- seqax = sequence modeling + JAX☆187Jul 23, 2025Updated 7 months ago
- Jax implementation of VIT-VQGAN☆10Jan 25, 2024Updated 2 years ago
- EasyRLHF aims to provide an easy and minimal interface to train aligned language models, using off-the-shelf solutions and datasets☆10Dec 12, 2023Updated 2 years ago
- Implicit Differentiable Optimal Control (IDOC) with JAX☆12May 11, 2022Updated 3 years ago
- A scalable benchmark for state representation learning in visual reinforcement learning.☆16Jun 23, 2025Updated 8 months ago
- ☆44Jul 12, 2024Updated last year
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Nov 23, 2023Updated 2 years ago
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated last year
- ☆52Jan 20, 2023Updated 3 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Jun 30, 2020Updated 5 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆82Mar 9, 2023Updated 3 years ago
- ☆31Jun 21, 2024Updated last year
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆89Nov 19, 2023Updated 2 years ago
- ☆13Jun 13, 2023Updated 2 years ago
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- Revisiting Peng's Q(lambda) for Modern Reinforcement Learning☆15Jul 23, 2021Updated 4 years ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆26Jan 14, 2025Updated last year
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- JAX implementation of the T5 model: Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer☆24Jun 10, 2023Updated 2 years ago
- official implementation of "CLIP-VQDiffusion : Langauge Free Training of Text To Image generation using CLIP and vector quantized diffusi…☆18Sep 5, 2024Updated last year
- An R Package for Bayesian Nonparametric Clustering. We plan to implement several models.☆16Jun 30, 2022Updated 3 years ago
- ☆14May 7, 2021Updated 4 years ago
- ☆18Jun 26, 2023Updated 2 years ago
- ☆18Feb 7, 2021Updated 5 years ago
- Open source code combining implementations of Upside Down Reinforcement Learning and Reward Conditioned Policies☆19Mar 10, 2021Updated 4 years ago
- The code of paper "Toward Optimal LLM Alignments Using Two-Player Games".☆17Jun 20, 2024Updated last year
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 3 months ago
- How to train a neural ODE for time series/weather forecasting☆39Feb 6, 2023Updated 3 years ago
- Official Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆24Aug 29, 2024Updated last year
- Efficient Video Prediction via Sparsely Conditioned Flow Matching. In ICCV, 2023.☆23Jun 5, 2024Updated last year
- Implementations of Stable Contrastive RL☆23Apr 13, 2025Updated 10 months ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆84Jul 27, 2022Updated 3 years ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆82May 13, 2024Updated last year