PrototypeCommon
willisma/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.
- Owner
- willisma (personal account)
- Stars
- Stars: 156
- Forks
- Forks: 14
- Language
- Python
- License
- MIT
- Created
- 2025
- Tier
- Prototype (100 to 249 ★) What this tier means
More repos by willisma
All 2 repos by willisma- willisma/SiTStars: 1,225PythonBeta
Nearby in Python
All Python repos- dengyie/zcode2apiStars: 156PythonPrototype
- SecFathy/xss-specialistStars: 156PythonPrototype
- mikuta0407/win4wineStars: 156PythonPrototype
- effigies/BitTornadoStars: 156PythonPrototype
- nipy/mindboggleStars: 156PythonPrototype
- apavlo/py-tpccStars: 156PythonPrototype