Input shape inference and SOTA custom layers for PyTorch.
Homepage PyPI
pip install torchlayers-nightly==1657760476
Login to resync this project