Gumbel Softmax Loss Function Guide and How to Implement It in PyTorch | Hacker News Reader