I would love to contribute. Did you start a repository somewhere?
For the implementation of basic probability distribution, is it possible to just wrap the Scipy function: https://github.com/pytorch/tutorials/blob/master/Creating%20extensions%20using%20numpy%20and%20scipy.ipynb 210