Rare find

diffuse_nnx. A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its variants as the primary backbone with support for ImageNet training and various sampling strategies.

github.com/willisma/diffuse_nnx

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.