Alternatives and similar repositories for jax-mmdit:

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

• cloneofsimo / min-max-in-dit
  ☆26Updated 9 months ago
• cloneofsimo / efae
  ☆21Updated 7 months ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆65Updated last month
• cloneofsimo / repa-rf
  ☆32Updated 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 6 months ago
• cloneofsimo / zeroshampoo
  ☆33Updated 5 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆36Updated last month
• cloneofsimo / project_RF
  ☆25Updated 8 months ago
• cloneofsimo / minDinoV2
  ☆17Updated 4 months ago
• apple / ml-ademamix
  ☆54Updated 3 months ago
• SerChirag / rs-imle
  RS-IMLE
  ☆38Updated 2 months ago
• cloneofsimo / karras-power-ema-tutorial
  ☆51Updated last year
• crowsonkb / consistency-models
  A JAX implementation of the continuous time formulation of Consistency Models
  ☆84Updated last year
• cloneofsimo / minSAE
  ☆24Updated 2 months ago
• ethansmith2000 / TransformerExperiments
  ☆19Updated 4 months ago
• patil-suraj / stable-diffusion-jax
  ☆86Updated 2 years ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆23Updated 3 weeks ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆26Updated 6 months ago
• Birch-san / imagebind-guided-diffusion
  Guide diffusion on ImageBind embedding similarity
  ☆28Updated last year
• ai-compiler-study / triton-kernels
  Triton kernels for Flux
  ☆19Updated last month
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆68Updated last year
• nreHieW / minVAR
  Minimal Implementation of Visual Autoregressive Modelling (VAR)
  ☆25Updated last month
• Birch-san / diffusers-play
  Repository with which to explore k-diffusion and diffusers, and within which changes to said packages may be tested.
  ☆54Updated last year
• ClashLuke / SOAP
  ☆21Updated 3 months ago
• timudk / flux_triton
  Writing FLUX in Triton
  ☆32Updated 4 months ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆96Updated 3 weeks ago
• yandex-research / vqdm
  Official repository for VQDM:Accurate Compression of Text-to-Image Diffusion Models via Vector Quantization paper
  ☆32Updated 4 months ago
• fal-ai-community / minDDPD
  ☆31Updated last month
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆76Updated 2 months ago