torchlayers (pypi): Input shape inference and SOTA custom layers for PyTorch.
Keywords: pytorch|keras|input|inference|automatic|shape|layers|sota|custom|imagenet|resnet|efficientnet