Alternatives and similar repositories for jax-fid-parallel

Users that are interested in jax-fid-parallel are comparing it to the libraries listed below

• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆39Updated last year
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆29Updated last year
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 3 years ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 10 months ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase
  ☆111Updated 3 weeks ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• google-research / sunds
  ☆26Updated last year
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆96Updated 3 years ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• gregorbachmann / scaling_mlps
  ☆52Updated last year
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆56Updated 2 years ago
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆32Updated 2 years ago
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆82Updated 3 years ago
• jiasenlu / LL3M
  LL3M: Large Language and Multi-Modal Model in Jax
  ☆74Updated last year
• orybkin / sigma-vae
  Repository for the paper "Simple and Effective VAE Training with Calibrated Decoders"
  ☆29Updated 4 years ago
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆57Updated last year
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆32Updated 2 years ago
• hyenal / relate
  Official PyTorch implementation of 'RELATE: Physically Plausible Multi-Object SceneSynthesis Using Structured Latent Spaces'.
  ☆31Updated 5 years ago
• kampta / PatchGame
  PyTorch implementation of "PatchGame: Learning to Signal Mid-level Patches in Referential Games" to appear in NeurIPS 2021
  ☆23Updated 4 years ago
• google-deepmind / affordances_option_models
  ☆23Updated 4 years ago
• jiasenlu / vit-vqgan-jax
  Jax implementation of VIT-VQGAN
  ☆10Updated last year
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆81Updated 4 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 4 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆115Updated 3 years ago
• CLAIRE-Labo / flash_attention
  A basic pure pytorch implementation of flash attention
  ☆16Updated last year
• haoliuhl / tux
  Tools and Utils for Experiments (TUX)
  ☆15Updated 9 months ago
• google-research / fitvid
  ☆66Updated 4 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆45Updated 2 years ago
• danijar / portal
  Fast and reliable distributed systems in Python
  ☆32Updated 7 months ago