Python Package optimised for Neural Network implementations