3 ms·Implementation of a Diffusion Transformer and Rectified Flow Sampling in Jax1 points by roborovskis 2y ago