Implementation of a Diffusion Transformer and Rectified Flow Sampling in Jax #1 Post by roborovskis » Mon, Jun 10, 2024, 1:02 AM UTC Implementation of a Diffusion Transformer and Rectified Flow Sampling in Jaxgithub.com