TAO Toolkit deep learning networks with PyTorch backend - View it on GitHub