Alternatives and similar repositories for FlaxDiff

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

• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆90Updated 9 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months ago
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆20Updated last year
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆35Updated 9 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
  ☆77Updated 9 months ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆61Updated 2 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• probabilists / azula
  Diffusion models in PyTorch
  ☆101Updated last month
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 4 months ago
• cgarciae / simple_flow_matching
  ☆23Updated 5 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 10 months ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆72Updated 4 months ago
• radarFudan / mamba-minimal-jax
  ☆30Updated 5 months ago
• apple / ml-agm
  ☆45Updated last year
• jax-ml / jax-llm-examples
  ☆109Updated this week
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 6 months ago
• brianfitzgerald / jax-mmdit
  Implementation of Diffusion Transformers and Rectified Flow in Jax
  ☆23Updated 10 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆80Updated 5 months ago
• yuanzhi-zhu / mini_edm
  Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist
  ☆52Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆82Updated 3 weeks ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆127Updated last year
• J-zin / Awesome-Diffusion-Flow-Samplers
  Collecting research materials on neural samplers with diffusion/flow models
  ☆46Updated 2 months ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆59Updated 3 weeks ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 3 months ago
• vdutor / neural-diffusion-processes
  Neural Diffusion Processes
  ☆77Updated 9 months ago
• AaronHavens / FlowMatching
  A repo where I play with conditional flow approaches for learning time-varying vector-fields.
  ☆18Updated 11 months ago
• smsharma / consistency-models
  Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.
  ☆20Updated last year
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆29Updated 11 months ago