FID computation in Jax/Flax.
☆29Jul 17, 2024Updated last year
Alternatives and similar repositories for jax-fid
Users that are interested in jax-fid are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Frechet inception distance (FID) evaluation in JAX☆14May 28, 2024Updated last year
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆17Aug 1, 2022Updated 3 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- ☆12Sep 10, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Learning to Skip the Middle Layers of Transformers☆17Aug 7, 2025Updated 9 months ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆264Mar 21, 2025Updated last year
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- A Python package designed for research in diffusion-based generative modeling☆32Nov 26, 2025Updated 5 months ago
- This Repository allows to super fast download historical ohlcv data from binance.☆12Nov 27, 2020Updated 5 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆120Jun 5, 2022Updated 3 years ago
- ☆16Jun 25, 2022Updated 3 years ago
- General Modules for JAX☆73Apr 7, 2026Updated last month
- Tensorflow 2.x implementation of Gradient Origin Networks☆12Jul 13, 2020Updated 5 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆46Jan 11, 2025Updated last year
- Unofficial JAX implementations of deep learning research papers☆162Jun 25, 2022Updated 3 years ago
- ☆24Jul 31, 2023Updated 2 years ago
- ☆67Jun 25, 2021Updated 4 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆63Jun 17, 2024Updated last year
- A GPT, made only of MLPs, in Jax☆59Jun 23, 2021Updated 4 years ago
- ☆21Apr 30, 2026Updated last week
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago
- Scalable metrics logging and analysis☆18Apr 9, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆28Mar 30, 2026Updated last month
- PIX is an image processing library in JAX, for JAX.☆436Mar 6, 2025Updated last year
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆25Aug 16, 2023Updated 2 years ago
- Official implementation of: "PlaySlot: Learning Inverse Latent Dynamics for Controllable Object-Centric Video Prediction and Planning" by…☆19Apr 1, 2026Updated last month
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆11Nov 23, 2023Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆111Jul 19, 2024Updated last year
- WBSR: Rethinking Imbalance in Image Super-Resolution for Efficient Inference☆13Oct 8, 2024Updated last year
- ☆14Aug 22, 2025Updated 8 months ago
- PysynthMRI is a python open-source tool which provides a user-friendly interface to synthesize contrast weighted images.☆25Nov 10, 2025Updated 5 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- ☆12Dec 9, 2022Updated 3 years ago
- CLU lets you write beautiful training loops in JAX.☆367Mar 3, 2026Updated 2 months ago
- [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification☆31Mar 30, 2025Updated last year
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- ☆32Oct 3, 2023Updated 2 years ago
- SuperPascal modification - compiles under fpc (Free Pascal).☆18Jun 30, 2018Updated 7 years ago
- Contains scripts for the PSI competition.☆10Dec 11, 2023Updated 2 years ago