Alternatives and similar repositories for jax-fid-parallel

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

• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆29Updated last year
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 11 months ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 3 years ago
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆82Updated 3 years ago
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆32Updated 2 years ago
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆97Updated 3 years ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆57Updated last year
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆59Updated 2 years ago
• kampta / PatchGame
  PyTorch implementation of "PatchGame: Learning to Signal Mid-level Patches in Referential Games" to appear in NeurIPS 2021
  ☆24Updated 4 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• gregorbachmann / scaling_mlps
  ☆52Updated last year
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase.
  ☆116Updated last month
• fattorib / transformer_shmap
  Tensor Parallelism with JAX + Shard Map
  ☆11Updated 2 years ago
• sholtodouglas / scalingExperiments
  ☆62Updated 3 years ago
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆81Updated 4 years ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 3 years ago
• crowsonkb / cloob-training
  CLOOB training (JAX) and inference (JAX and PyTorch)
  ☆74Updated 3 years ago
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆70Updated 2 years ago
• HomebrewML / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆43Updated 2 years ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆46Updated 2 years ago
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆57Updated 2 years ago
• kvfrans / splus
  ☆122Updated 6 months ago
• cloneofsimo / scaling-guide
  WIP
  ☆93Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆118Updated 3 years ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆63Updated last week
• Popgun-Labs / PopGen
  A generative modelling toolkit for PyTorch.
  ☆70Updated 4 years ago
• Newbeeer / Anytime-Auto-Regressive-Model
  Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"
  ☆26Updated 2 years ago