matthias-wright / jax-fid

Related projects ⓘ

Alternatives and complementary repositories for jax-fid

• cloneofsimo / zeroshampoo
  ☆31Updated 2 months ago
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆77Updated 2 years ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated last year
• layer6ai-labs / calo-forest
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆17Updated 2 weeks ago
• crowsonkb / consistency-models
  A JAX implementation of the continuous time formulation of Consistency Models
  ☆83Updated last year
• Newbeeer / stf
  Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Models”
  ☆68Updated last year
• lucidrains / logavgexp-torch
  Implementation of LogAvgExp for Pytorch
  ☆32Updated 2 years ago
• gisilvs / AEF
  ☆33Updated last year
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆16Updated this week
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆30Updated last year
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated last year
• cloneofsimo / efae
  ☆21Updated 5 months ago
• ClashLuke / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆39Updated last year
• gregorbachmann / scaling_mlps
  ☆51Updated 5 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆32Updated last year
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆37Updated 2 years ago
• Newbeeer / Anytime-Auto-Regressive-Model
  Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"
  ☆24Updated last year
• Jack000 / guided-diffusion
  ☆28Updated 2 years ago
• cloneofsimo / karras-power-ema-tutorial
  ☆50Updated 10 months ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆52Updated last year
• cloneofsimo / repa-rf
  ☆27Updated 2 weeks ago
• nshepperd / jaxtorch
  A JAX nn library
  ☆21Updated 8 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆24Updated last week
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆57Updated last year
• LAION-AI / laion50BU
  Un-*** 50 billions multimodality dataset
  ☆24Updated 2 years ago
• crowsonkb / cloob-training
  CLOOB training (JAX) and inference (JAX and PyTorch)
  ☆70Updated 2 years ago