An attempt at a pytorch codebase I'd like. - View it on GitHub