jiasenlu / vit-vqgan-jaxLinks
Jax implementation of VIT-VQGAN
☆10Updated last year
Alternatives and similar repositories for vit-vqgan-jax
Users that are interested in vit-vqgan-jax are comparing it to the libraries listed below
Sorting:
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆26Updated 2 years ago
- "Learning Discrete and Continuous Factors of Data via Alternating Disentanglement" accepted at ICML2019☆21Updated 6 years ago
- [ICLR2024] (EvALign-ICL Benchmark) Beyond Task Performance: Evaluating and Reducing the Flaws of Large Multimodal Models with In-Context …☆22Updated last year
- Generalised UDRL☆37Updated 3 years ago
- Solving reinforcement learning tasks which require language and vision☆33Updated 2 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- ☆24Updated 4 years ago
- Source code for paper Conservative Uncertainty Estimation By Fitting Prior Networks (ICLR 2020)☆21Updated 2 years ago
- ☆20Updated 5 years ago
- PyTorch code for "Perceiver-VL: Efficient Vision-and-Language Modeling with Iterative Latent Attention" (WACV 2023)☆33Updated 2 years ago
- An adaptive training algorithm for residual network☆17Updated 5 years ago
- ReaSCAN is a synthetic navigation task that requires models to reason about surroundings over syntactically difficult languages. (NeurIPS…☆19Updated 3 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 4 years ago
- PyTorch implementation of "PatchGame: Learning to Signal Mid-level Patches in Referential Games" to appear in NeurIPS 2021☆23Updated 4 years ago
- Code for "MIM: Mutual Information Machine" paper.☆16Updated 2 years ago
- AdaCat☆49Updated 3 years ago
- ☆31Updated last year
- ☆30Updated 3 years ago
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Updated 5 years ago
- Repository for the paper "Simple and Effective VAE Training with Calibrated Decoders"☆29Updated 4 years ago
- ☆42Updated last year
- A discrete sequential VAE☆40Updated 5 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- Official code for paper "Non-Adversarial Image Synthesis with Generative Latent Nearest Neighbors"☆28Updated 5 years ago
- ☆27Updated 5 years ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆65Updated 2 years ago
- Representation learning for NLP @ JSALT19☆39Updated 5 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆49Updated 2 months ago
- This repository hosts the dataset and source code for "A causal view of compositional zero-shot recognition". Yuval Atzmon, Felix Kreuk, …☆27Updated 4 years ago
- Official code for the paper "Compositional Generalization from First Principles" (NeurIPS 2023)☆13Updated 2 years ago