Some basic Neural Network stuff done with numpy and pytorch