pcuenca / lpips-j
Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the π€ Hugging Face hub.
β15Updated 2 years ago
Alternatives and similar repositories for lpips-j:
Users that are interested in lpips-j are comparing it to the libraries listed below
- β33Updated last year
- FID computation in Jax/Flax.β27Updated 7 months ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paperβ104Updated 3 years ago
- JAX implementation ViT-VQGANβ80Updated 2 years ago
- β44Updated 9 months ago
- β35Updated last year
- β33Updated 5 months ago
- β86Updated 2 years ago
- β51Updated last year
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"β51Updated 9 months ago
- β49Updated 3 years ago
- The 2D discrete wavelet transform for JAXβ40Updated last year
- Easy Hypernetworks in Pytorch and Jaxβ97Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"β25Updated last year
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adamβ73Updated 6 months ago
- β30Updated 3 months ago
- AdaCatβ49Updated 2 years ago
- Official code for the paper "Image generation with shortest path diffusion" accepted at ICML 2023.β22Updated last year
- Very deep VAEs in JAX/Flaxβ46Updated 3 years ago
- Implementation of LogAvgExp for Pytorchβ33Updated 2 years ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integratorβ156Updated 2 years ago
- β26Updated 9 months ago
- β21Updated 8 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)β157Updated last year
- A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.β71Updated 3 years ago
- Code for https://arxiv.org/abs/2406.04329β58Updated 2 months ago
- NF-Layers for constructing neural functionals.β82Updated last year
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMCβ137Updated 4 months ago
- A repository for the updated version of CoinRun used to collect MUGEN, a multimodal video-audio-text dataset. This repo contains scripts β¦β13Updated 2 years ago
- The Official PyTorch Implementation of "VAEBM: A Symbiosis between Variational Autoencoders and Energy-based Models" (ICLR 2021 spotlightβ¦β54Updated 2 years ago