Experimental ground for optimizing memory of pytorch models - View it on GitHub