PyTorch upgrade

[brownadder]

All computations need to be done using PyTorch instead of numpy.

Need to identify all operations which are being done using numpy, and convert.