Is there something similar for diffusion models? By the way, this is incredibly useful for learning in depth the core of LLM's.