storchastic (pypi): Stochastic Deep Learning for PyTorch
Keywords: gradient-estimators|pytorch|reparameterization|stochastic-computation-graphs|stochastic-deep-learning