pcuenca / lpips-j
Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the π€ Hugging Face hub.
β14Updated 2 years ago
Related projects β
Alternatives and complementary repositories for lpips-j
- FID computation in Jax/Flax.β24Updated 3 months ago
- β42Updated 6 months ago
- JAX implementation ViT-VQGANβ77Updated 2 years ago
- β33Updated last year
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Modelsββ68Updated last year
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAIβ84Updated 2 years ago
- β32Updated 9 months ago
- My take on Flow Matchingβ17Updated this week
- Efficient optimizersβ58Updated this week
- A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.β71Updated 2 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)β157Updated last year
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"β48Updated 6 months ago
- β86Updated 2 years ago
- β50Updated 10 months ago
- Implementation of LogAvgExp for Pytorchβ32Updated 2 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library β€οΈβ52Updated last year
- β50Updated 5 months ago
- CLOOB training (JAX) and inference (JAX and PyTorch)β70Updated 2 years ago
- β31Updated 2 months ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"β24Updated last year
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonettoβ53Updated 5 months ago
- β49Updated 3 years ago
- Un-*** 50 billions multimodality datasetβ24Updated 2 years ago
- β21Updated 4 months ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnistβ38Updated 10 months ago
- Transformer implementation for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"β55Updated 3 weeks ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integratorβ154Updated last year
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variationβ12Updated 3 years ago
- AdaCatβ49Updated 2 years ago