matthias-wright / jax-fidView external linksLinks
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 6 months ago
- Frechet inception distance (FID) evaluation in JAX☆14May 28, 2024Updated last year
- Tensorflow 2.x implementation of Gradient Origin Networks☆12Jul 13, 2020Updated 5 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
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆265Mar 21, 2025Updated 10 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
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 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
- Implementation of Direct Preference Optimization☆17Jul 17, 2023Updated 2 years ago
- ☆16Jun 25, 2022Updated 3 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Jan 11, 2025Updated last year
- JAX port of FLUX.1 models using flax.nnx☆24Sep 28, 2024Updated last year
- ☆24Jul 31, 2023Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆110Jul 19, 2024Updated last year
- ☆25Apr 10, 2025Updated 10 months ago
- 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.☆433Mar 6, 2025Updated 11 months ago
- General Modules for JAX☆72Sep 12, 2025Updated 5 months 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
- ☆33Nov 4, 2024Updated last year
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Mar 23, 2022Updated 3 years ago
- Adaptive wavelet pooling for CNN in PyTorch, AISTATS 2021.☆32Sep 20, 2021Updated 4 years ago
- Using advances in generative modeling to learn reward functions from unlabeled videos.☆140Feb 12, 2024Updated 2 years ago
- ☆32Oct 3, 2023Updated 2 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
- Reinforcement learning environment for UR5e robot with OPENAI gym like format. Include both simulation and real parts.☆14Nov 2, 2021Updated 4 years ago
- This repo contains all the code related to the interactive online Bloch sphere simulator☆18Jan 19, 2026Updated 3 weeks ago
- An official implementation of the paper "Unsupervised Keypoint Learning for Guiding Class-conditional Video Prediction", NeurIPS 2019☆30Nov 22, 2022Updated 3 years ago
- [3DV 2020] PeeledHuman: Robust Shape Representation for Textured 3D Human Body Reconstruction☆41May 21, 2021Updated 4 years ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Dec 8, 2022Updated 3 years ago
- 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
- A data processing module implemented with numpy☆10Aug 16, 2022Updated 3 years ago
- ☆20Sep 5, 2025Updated 5 months ago
- An Awesome list of AI tools powered by ChatGPT / Whisper and Stable DIffusion or are useful to developers of that domain☆10Jul 26, 2023Updated 2 years ago
- Reference implementation of Thin and Deep Gaussian Processes (NeurIPS 2023)☆13Nov 25, 2024Updated last year