Size inference and pipeline notation for PyTorch models. - View it on GitHub