https://github.com/vachanvy/diffusion-transformer

Pytorch and JAX Implementation of Scalable Diffusion Models with Transformers | Diffusion Transformers in Pytorch and JAX

deep-learning diffusion-models diffusion-transformer jax jax-implementation python pytorch pytorch-implementation transformers

Last synced: 10 Mar 2026