G github.com Implementation of a Diffusion Transformer and Rectified Flow Sampling in Jax 1 points roborovskis 2 years ago 0 comments