Alternatives and similar repositories for FlaxDiff:

Users that are interested in FlaxDiff are comparing it to the libraries listed below

• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆26Updated 9 months ago
• tung-nd / TNP-pytorch
  Official implementation of Transformer Neural Processes
  ☆71Updated 2 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆41Updated 2 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆86Updated 7 months ago
• dasayan05 / iclr24_blog_code
  Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
  ☆18Updated last year
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆32Updated 7 months ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 2 years ago
• AlexiaJM / score_sde_fast_sampling
  Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper
  ☆104Updated 3 years ago
• yuanzhi-zhu / mini_edm
  Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist
  ☆50Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 3 months ago
• cloneofsimo / ezmup
  Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam
  ☆73Updated 8 months ago
• louaaron / Reflected-Diffusion
  [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)
  ☆157Updated last year
• radarFudan / mamba-minimal-jax
  ☆30Updated 4 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 5 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated last year
• antonioterpin / jkonet-star
  JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
  ☆30Updated 2 weeks ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated 9 months ago
• yang-song / score_flow
  Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)
  ☆141Updated 3 years ago
• sbyebss / variational_wgf
  [ICML2022] Variational Wasserstein gradient flow
  ☆21Updated 2 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆30Updated 3 months ago
• UW-Madison-Lee-Lab / SFT-PG
  Code for "Optimizing DDPM Sampling with Shortcut Fine-Tuning" (https://arxiv.org/abs/2301.13362), ICML 2023
  ☆29Updated last year
• sangyun884 / fast-ode
  Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023
  ☆81Updated last month
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆45Updated 2 years ago
• giannisdaras / ambient-tweedie
  [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"
  ☆53Updated 11 months ago
• enhuiz / langevin-dynamics-and-score-matching
  A demo shows how to combine Langevin dynamics with score matching for generative models.
  ☆37Updated 4 years ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆77Updated 5 months ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆99Updated last year
• cloneofsimo / zeroshampoo
  ☆33Updated 6 months ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆147Updated 2 years ago