GitHub repository: HEmile/storchastic
Primary language: Python
Description: Stochastic Automatic Differentiation library for PyTorch.