A numerically-stable and differentiable implementation of the Truncated Gaussian distribution in Pytorch.