pytorch version of the SAFE neural network - View it on GitHub