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
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆31Updated 2 years ago
- Source code for paper Conservative Uncertainty Estimation By Fitting Prior Networks (ICLR 2020)☆21Updated 2 years ago
- Solving reinforcement learning tasks which require language and vision☆33Updated 2 years ago
- AdaCat☆49Updated 3 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆48Updated last week
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆80Updated 3 years ago
- ☆25Updated last year
- "Learning Discrete and Continuous Factors of Data via Alternating Disentanglement" accepted at ICML2019☆21Updated 6 years ago
- ReaSCAN is a synthetic navigation task that requires models to reason about surroundings over syntactically difficult languages. (NeurIPS…☆20Updated 3 years ago
- Code for "MIM: Mutual Information Machine" paper.☆16Updated 2 years ago
- ☆23Updated 3 years ago
- ☆20Updated 5 years ago
- Generalised UDRL☆37Updated 3 years ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆60Updated 2 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- An adaptive training algorithm for residual network☆17Updated 5 years ago
- Reimplementation code for the paper "Generative Temporal Models with Spatial Memory for Partially Observed Environments"☆29Updated 3 years ago
- ☆24Updated 3 years ago
- Hybrid Discriminative-Generative Training via Contrastive Learning☆75Updated 2 years ago
- ☆32Updated 4 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- ☆41Updated last year
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆105Updated 3 years ago
- Implementation of Memformer, a Memory-augmented Transformer, in Pytorch☆121Updated 4 years ago
- ☆32Updated last year
- ☆50Updated 4 years ago
- Implementation of Kronecker Attention in Pytorch☆19Updated 4 years ago