kvfrans / jax-fid-parallelLinks
Frechet inception distance (FID) evaluation in JAX
☆12Updated last year
Alternatives and similar repositories for jax-fid-parallel
Users that are interested in jax-fid-parallel are comparing it to the libraries listed below
Sorting:
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆34Updated last year
- A simple, performant and scalable JAX-based world modeling codebase☆70Updated this week
- GPT implementation in Flax☆18Updated 3 years ago
- ☆121Updated 6 months ago
- Fast and reliable distributed systems in Python☆28Updated 4 months ago
- 👆PyTorch Implementation of JEDi Metric described in "Beyond FVD: Enhanced Evaluation Metrics for Video Generation Quality"☆27Updated 8 months ago
- FID computation in Jax/Flax.☆28Updated last year
- ☆26Updated last year
- MineRL Navigate Video Dataset☆12Updated 4 years ago
- ☆115Updated 2 months ago
- An open source implementation of the gameNgen paper☆41Updated 3 months ago
- WIP☆94Updated last year
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆82Updated 7 months ago
- Flow-matching algorithms in JAX☆104Updated last year
- Clockwork Variational Autoencoder☆47Updated 4 years ago
- EfficientMORL (ICML'21)☆21Updated 3 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- Tools and Utils for Experiments (TUX)☆15Updated 7 months ago
- ☆35Updated 3 years ago
- Official PyTorch implementation of 'RELATE: Physically Plausible Multi-Object SceneSynthesis Using Structured Latent Spaces'.☆31Updated 4 years ago
- Jax implementation of VIT-VQGAN☆10Updated last year
- This is the official source code for SLATE. We provide the code for the model, the training code, and a dataset loader for the 3D Shapes …☆86Updated 2 years ago
- A simple library for scaling up JAX programs☆143Updated 9 months ago
- ☆63Updated last year
- ☆27Updated 3 months ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 3 months ago
- Open source code for paper "Denoised MDPs: Learning World Models Better Than the World Itself"☆134Updated 2 years ago