HEmile / storchastic

Stochastic Automatic Differentiation library for PyTorch.
GNU General Public License v3.0
180 stars 5 forks source link

Add functionality for instantiating all possible methods for some distribution #73

Open HEmile opened 4 years ago

HEmile commented 4 years ago

Using rsample, is_enumerable and other continuity checks we can figure out what methods are suitable to what problem. We should add functionality that lists all the methods that can be used. Example: storch.get_methods(OneHotCategorical, is_differentiable=False, n=8) or something like this.