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