kvfrans / jax-fid-parallel
Frechet inception distance (FID) evaluation in JAX
☆12Updated 8 months ago
Alternatives and similar repositories for jax-fid-parallel:
Users that are interested in jax-fid-parallel are comparing it to the libraries listed below
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆24Updated 8 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆52Updated 3 weeks ago
- GPT implementation in Flax☆18Updated 3 years ago
- ☆25Updated 8 months ago
- ☆59Updated 7 months ago
- Configuration with Dataclasses+YAML+Argparse. Fork of Pyrallis☆30Updated 2 weeks ago
- ☆51Updated 8 months ago
- Bingham Policy Parameterization for 3D Rotations in Reinforcement Learning☆24Updated 3 years ago
- Fast and reliable distributed systems in Python☆24Updated this week
- WIP☆93Updated 6 months ago
- FID computation in Jax/Flax.☆26Updated 7 months ago
- ☆11Updated 7 months ago
- ☆30Updated 3 years ago
- Flow-matching algorithms in JAX☆83Updated 6 months ago
- ☆31Updated 10 months ago
- Official implementation of the paper "EgoPet: Egomotion and Interaction Data from an Animal's Perspective".☆25Updated 7 months ago
- [TMLR 2024] Official PyTorch Implementation of Deep Dynamic Latent Particles☆16Updated last year
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆11Updated 11 months ago
- ☆23Updated last week
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆47Updated 2 months ago
- Tools and Utils for Experiments (TUX)☆15Updated 3 weeks ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆35Updated 2 years ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆92Updated 4 months ago
- ☆116Updated last year
- 👆PyTorch Implementation of JEDi Metric described in "Beyond FVD: Enhanced Evaluation Metrics for Video Generation Quality"☆18Updated 2 months ago
- Clockwork Variational Autoencoder☆46Updated 3 years ago
- flexible meta-learning in jax☆12Updated last year
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16Updated last year
- PyTorch Package For Quasimetric Learning☆41Updated 3 months ago