Check out torchhgg (@torchdag) on Linkme