pcuenca / lpips-jLinks
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
Sorting:
- β33Updated 2 years ago
- FID computation in Jax/Flax.β27Updated 11 months ago
- The 2D discrete wavelet transform for JAXβ43Updated 2 years ago
- JAX implementation ViT-VQGANβ83Updated 2 years ago
- β44Updated last year
- Implementation of LogAvgExp for Pytorchβ36Updated 2 months ago
- β34Updated 9 months ago
- β89Updated 2 years ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paperβ104Updated 3 years ago
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Modelsββ76Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"β26Updated 2 years ago
- β51Updated 4 years ago
- A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.β73Updated 3 years ago
- β61Updated 2 years ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorchβ25Updated 5 months ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variationβ12Updated 4 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)β157Updated last year
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integratorβ157Updated 2 years ago
- Code for the paper: Rotating Features for Object Discoveryβ52Updated 10 months ago
- β40Updated last year
- AdaCatβ49Updated 2 years ago
- β19Updated last month
- JAX Implementation of Black Forest Labs' Flux.1 family of modelsβ34Updated 8 months ago
- β23Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021β27Updated 3 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Modelsβ45Updated 2 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library β€οΈβ54Updated 2 years ago
- Implementation of a holodeck, written in Pytorchβ18Updated last year
- Implementation of Retrieval-Augmented Denoising Diffusion Probabilistic Models in Pytorchβ64Updated 3 years ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adamβ80Updated 10 months ago