brianfitzgerald/jax-mmdit

Implementation of Diffusion Transformers and Rectified Flow in Jax

1 best points · 0 comments · 1 Hacker News post · 31 GitHub stars · View on GitHub

Hacker News discussions

• Implementation of a Diffusion Transformer and Rectified Flow Sampling in Jax (1 points, 0 comments, by roborovskis, 2024-06-10)