brianfitzgerald / jax-mmdit
Implementation of Diffusion Transformers and Rectified Flow in Jax
☆21Updated 8 months ago
Alternatives and similar repositories for jax-mmdit:
Users that are interested in jax-mmdit are comparing it to the libraries listed below
- ☆21Updated 9 months ago
- ☆27Updated 10 months ago
- ☆32Updated 4 months ago
- FID computation in Jax/Flax.☆27Updated 8 months ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆73Updated 7 months ago
- RS-IMLE☆38Updated 3 months ago
- ☆25Updated 9 months ago
- ☆33Updated 6 months ago
- Focused on fast experimentation and simplicity☆69Updated 2 months ago
- ☆58Updated 4 months ago
- A JAX implementation of the continuous time formulation of Consistency Models☆84Updated last year
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆55Updated 10 months ago
- ☆24Updated 3 months ago
- Official repository for VQDM:Accurate Compression of Text-to-Image Diffusion Models via Vector Quantization paper☆33Updated 6 months ago
- My take on Flow Matching☆41Updated 2 months ago
- ☆30Updated 3 months ago
- The 2D discrete wavelet transform for JAX☆41Updated 2 years ago
- Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)☆44Updated 4 months ago
- ☆45Updated 10 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 5 months ago
- ☆19Updated 5 months ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆29Updated 5 months ago
- Utilities for PyTorch distributed☆23Updated 3 weeks ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated 2 months ago
- Repository with which to explore k-diffusion and diffusers, and within which changes to said packages may be tested.☆55Updated last year
- ☆43Updated 3 weeks ago