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
- FID computation in Jax/Flax.β27Updated 8 months ago
- β33Updated 2 years ago
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Modelsββ73Updated last year
- β88Updated 2 years ago
- JAX implementation ViT-VQGANβ82Updated 2 years ago
- Easy Hypernetworks in Pytorch and Jaxβ98Updated 2 years ago
- The 2D discrete wavelet transform for JAXβ41Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"β26Updated last year
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paperβ104Updated 3 years ago
- β45Updated 10 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)β158Updated last year
- β36Updated last year
- Implementation of LogAvgExp for Pytorchβ34Updated 2 years ago
- Automatically take good care of your preemptible TPUsβ36Updated last year
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library β€οΈβ54Updated 2 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".β32Updated 2 years ago
- Codebase of Truncated Consistency Models (ICLR 2025)β19Updated 2 months ago
- β33Updated 6 months ago
- β22Updated 9 months ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integratorβ156Updated 2 years ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorchβ23Updated 2 months ago
- β51Updated last year
- NF-Layers for constructing neural functionals.β82Updated last year
- A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.β72Updated 3 years ago
- β50Updated 4 years ago
- β61Updated 2 years ago
- β30Updated 4 months ago
- Train vision models using JAX and π€ transformersβ96Updated 2 months ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adamβ73Updated 7 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAIβ86Updated 3 years ago