kvfrans / jax-fid-parallel
Frechet inception distance (FID) evaluation in JAX
☆12Updated 11 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
- 3D Gaussian Splatting in JAX☆57Updated 11 months ago
- Fast and reliable distributed systems in Python☆26Updated 3 weeks ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆28Updated 11 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- FID computation in Jax/Flax.☆27Updated 9 months ago
- 👆PyTorch Implementation of JEDi Metric described in "Beyond FVD: Enhanced Evaluation Metrics for Video Generation Quality"☆20Updated 5 months ago
- WIP☆93Updated 8 months ago
- Configuration with Dataclasses+YAML+Argparse. Fork of Pyrallis☆42Updated last week
- ☆26Updated 11 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆102Updated last month
- Focused on fast experimentation and simplicity☆72Updated 4 months ago
- ☆61Updated 9 months ago
- ☆116Updated 2 months ago
- ☆64Updated 10 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆62Updated 3 months ago
- PyTorch Package For Quasimetric Learning☆42Updated 6 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆24Updated 3 months ago
- ☆51Updated 11 months ago
- Minimal but scalable implementation of large language models in JAX☆34Updated 6 months ago
- A Video Tokenizer Evaluation Dataset☆114Updated 3 months ago
- Flow-matching algorithms in JAX☆90Updated 9 months ago
- Tools and Utils for Experiments (TUX)☆16Updated 3 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆98Updated 7 months ago
- [TMLR 2024] Official PyTorch Implementation of Deep Dynamic Latent Particles☆16Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 5 months ago
- research impl of Native Sparse Attention (2502.11089)☆53Updated 2 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆123Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated last year
- ☆11Updated 9 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆88Updated 6 months ago