A easy to use API to store outputs from forward/backward hooks in Pytorch - View it on GitHub