kvfrans / jax-vqvae-vqganView external linksLinks
JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
☆41Jun 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 last month
- Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"☆20Oct 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 6 months ago
- seqax = sequence modeling + JAX☆170Jul 23, 2025Updated 6 months 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
- Offline Policy Evaluation via Adaptive Weighting with Data from Contextual Bandits☆10Oct 21, 2024Updated last year
- Jax implementation of VIT-VQGAN☆10Jan 25, 2024Updated 2 years ago
- A scalable benchmark for state representation learning in visual reinforcement learning.☆16Jun 23, 2025Updated 7 months ago
- Implicit Differentiable Optimal Control (IDOC) with JAX☆12May 11, 2022Updated 3 years 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
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Jun 30, 2020Updated 5 years ago
- ☆20Dec 7, 2025Updated 2 months ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 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 2 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)☆22Jan 14, 2025Updated last year
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- official implementation of "CLIP-VQDiffusion : Langauge Free Training of Text To Image generation using CLIP and vector quantized diffusi…☆18Sep 5, 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
- Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆23Aug 29, 2024Updated last year
- The code of paper "Toward Optimal LLM Alignments Using Two-Player Games".☆17Jun 20, 2024Updated last year
- ☆18Feb 7, 2021Updated 5 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 2 months ago
- ☆18Jun 26, 2023Updated 2 years ago
- How to train a neural ODE for time series/weather forecasting☆39Feb 6, 2023Updated 3 years ago
- Implementations of Stable Contrastive RL☆23Apr 13, 2025Updated 10 months ago
- Efficient Video Prediction via Sparsely Conditioned Flow Matching. In ICCV, 2023.☆23Jun 5, 2024Updated last year
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 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
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Jul 6, 2023Updated 2 years ago
- Official implementation of "Approximating Gradients for Differentiable Quality Diversity in Reinforcement Learning"☆22Oct 3, 2022Updated 3 years ago