[lucidrains]

very nice, will have to try it out! this is the same research group from which Robin Rombach (of stable diffusion fame) originated from