GitHub repository: nomutin/distribution-extension
Primary language: Python
Description: Useful torch.distributions API