This is the code for 'Build an AI Startup with Pytorch" by Siraj Raval - View it on GitHub