Switch to pytorch

[mortonjt]

Pytorch is proving to be much much easier to debug.

This will be key, especially when enabling Bayesian inference via variational inference.

[mortonjt]

Issues #45 #49 #50 try to address this.