https://github.com/user-attachments/files/28985267/Gumbel.Distillation.pdf