LightKit provides simple utilities for working with PyTorch and PyTorch Lightning. At the moment, it provides three simple features:

• A data loader for tabular data that is orders of magnitude faster than PyTorch's builtin data loader for medium-sized datasets and larger ones.
• A mixin for modules that allows to save not only weights, but also the configuration to the file system such that it is easier to retrieve trained models.
• A typed base class for estimators that enables users to easily create estimators with PyTorch and PyTorch Lightning which are fully compatible with Scikit-learn.

For more details, consult the documentation.

LightKit is available via pip:

pip install lightkit

If you are using Poetry:

poetry add lightkit