Implementation of a Diffusion Transformer and Rectified Flow Sampling in Jax

No comments yet.