Deprecated in favor of Lightning Bolts - https://pypi.org/project/lightning-bolts
pip install pytorch-lightning-bolts