Parametrized types

[emerali]

At the moment, all of the models (well, actually everything) use the torch.double data-type by default, but it would be nice to have this be an optional argument to the model constructors.