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 and start building anything your business needs.
- ☆12Jul 6, 2023Updated 2 years ago
- JAX port of FLUX.1 models using flax.nnx☆24Sep 28, 2024Updated last year
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆265Mar 21, 2025Updated last year
- Implementation of Direct Preference Optimization☆17Jul 17, 2023Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆119Jun 5, 2022Updated 3 years ago
- ☆16Jun 25, 2022Updated 3 years ago
- Tensorflow 2.x implementation of Gradient Origin Networks☆12Jul 13, 2020Updated 5 years ago
- 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☆161Jun 25, 2022Updated 3 years ago
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- ☆24Jul 31, 2023Updated 2 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
- A repository for the updated version of CoinRun used to collect MUGEN, a multimodal video-audio-text dataset. This repo contains scripts …☆13Jul 13, 2022Updated 3 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- [3DV2026] Official repository for "CamC2V: Context-aware Controllable Video Generation"☆14Nov 11, 2025Updated 4 months ago
- ☆20Sep 5, 2025Updated 6 months ago
- 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 11 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.
- Pulsar-plot of Strava runs in Swift 3☆12Apr 2, 2018Updated 7 years ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 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
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Nov 23, 2023Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆111Jul 19, 2024Updated last year
- All Kaggle notebooks and datasets☆20Jun 13, 2024Updated last year
- ☆33Nov 4, 2024Updated last year
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Mar 23, 2022Updated 4 years ago
- OpenAI Gym compatible RL environments for deformable linear object manipulation.☆19Aug 4, 2022Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆12Dec 9, 2022Updated 3 years ago
- ☆32May 5, 2023Updated 2 years ago
- [CVPR2023] Practical Network Acceleration with Tiny Sets☆14Jul 28, 2023Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆367Mar 3, 2026Updated 3 weeks ago
- ☆32Oct 3, 2023Updated 2 years ago
- Contains scripts for the PSI competition.☆10Dec 11, 2023Updated 2 years ago
- [ICCV 2025] MPG-SAM 2: Adapting SAM 2 with Mask Priors and Global Context for Referring Video Object Segmentation☆22Sep 5, 2025Updated 6 months ago