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
- Source code for paper Conservative Uncertainty Estimation By Fitting Prior Networks (ICLR 2020)☆21Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆31Updated 2 years ago
- "Learning Discrete and Continuous Factors of Data via Alternating Disentanglement" accepted at ICML2019☆21Updated 5 years ago
- Solving reinforcement learning tasks which require language and vision☆33Updated 2 years ago
- An adaptive training algorithm for residual network☆15Updated 4 years ago
- [ICLR2024] (EvALign-ICL Benchmark) Beyond Task Performance: Evaluating and Reducing the Flaws of Large Multimodal Models with In-Context …☆22Updated last year
- ☆20Updated 5 years ago
- Z Forcing: Training Stochastic RNN's, NIPS'17☆32Updated 7 years ago
- ☆23Updated 3 years ago
- Official code for paper "Non-Adversarial Image Synthesis with Generative Latent Nearest Neighbors"☆28Updated 5 years ago
- STABILIZING GRADIENTS FOR DEEP NEURAL NETWORKS VIA EFFICIENT SVD PARAMETERIZATION☆16Updated 7 years ago
- GAN(TK)²: GAN Neural Tangent Kernel ToolKit☆13Updated 3 years ago
- ☆36Updated 4 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- Code for "MIM: Mutual Information Machine" paper.☆16Updated 2 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆51Updated 3 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆38Updated 4 years ago
- Official PyTorch code release for Implicit Gradient Transport, NeurIPS'19☆21Updated 6 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- ☆30Updated 3 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆63Updated 3 years ago
- The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …☆16Updated last month
- Generalised UDRL☆37Updated 3 years ago
- Hybrid Discriminative-Generative Training via Contrastive Learning☆75Updated 2 years ago
- MTAdam: Automatic Balancing of Multiple Training Loss Terms☆36Updated 4 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 4 years ago