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
Sorting:
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- Learning to Skip the Middle Layers of Transformers☆17Aug 7, 2025Updated 7 months ago
- Frechet inception distance (FID) evaluation in JAX☆14May 28, 2024Updated last year
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- Tensorflow 2.x implementation of Gradient Origin Networks☆12Jul 13, 2020Updated 5 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆265Mar 21, 2025Updated 11 months ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆17Aug 1, 2022Updated 3 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
- Implementation of Direct Preference Optimization☆17Jul 17, 2023Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆46Jan 11, 2025Updated last year
- Resa: Transparent Reasoning Models via SAEs☆47Sep 23, 2025Updated 5 months ago
- ☆25Apr 10, 2025Updated 10 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆111Jul 19, 2024Updated last year
- Unofficial JAX implementations of deep learning research papers☆161Jun 25, 2022Updated 3 years ago
- OpenAI Gym compatible RL environments for deformable linear object manipulation.☆19Aug 4, 2022Updated 3 years ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated last year
- General Modules for JAX☆72Feb 21, 2026Updated 2 weeks ago
- ☆67Jun 25, 2021Updated 4 years ago
- 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
- ☆13Oct 5, 2025Updated 5 months ago
- [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification☆31Mar 30, 2025Updated 11 months ago
- ☆32Oct 3, 2023Updated 2 years ago
- Using advances in generative modeling to learn reward functions from unlabeled videos.☆142Feb 12, 2024Updated 2 years ago
- Adaptive wavelet pooling for CNN in PyTorch, AISTATS 2021.☆32Sep 20, 2021Updated 4 years ago
- Build an AI bot in Discord to serve user's personalized reports on what's up in tech☆28Sep 14, 2025Updated 5 months ago
- An official implementation of the paper "Unsupervised Keypoint Learning for Guiding Class-conditional Video Prediction", NeurIPS 2019☆30Nov 22, 2022Updated 3 years ago
- Texture mapping with variational auto-encoders☆40Oct 1, 2021Updated 4 years ago
- [3DV 2020] PeeledHuman: Robust Shape Representation for Textured 3D Human Body Reconstruction☆41May 21, 2021Updated 4 years ago
- ☆20Sep 5, 2025Updated 6 months ago
- my profile readme☆14Updated this week
- Source codes of Learning Causal Representations for Robust Domain Adaptation (IEEE TKDE)☆12Feb 14, 2022Updated 4 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆38Jun 14, 2022Updated 3 years ago
- ☆34Oct 27, 2020Updated 5 years ago
- (CVPR 2026) Long-RVOS: A Comprehensive Benchmark for Long-term Referring Video Object Segmentation☆27Feb 28, 2026Updated last week
- Project focused on enhancing the quality of low-fidelity endoscopy images using Generative Adversarial Networks (GANs) implemented in PyT…☆17Jun 5, 2025Updated 9 months ago
- A mouse brain histology tool for neuroscientists.☆13Feb 16, 2026Updated 2 weeks ago
- Reference implementation of Thin and Deep Gaussian Processes (NeurIPS 2023)☆14Nov 25, 2024Updated last year
- Communication Relay by creating a WiFi Mesh Network using ROS, and using that network for Data Telemetry, with Telemetry radios ( Ubiquit…☆11Dec 18, 2018Updated 7 years ago