PyTorch Lightning

PyTorch Lightning

The light PyTorch wrapper for high-performance AI research

5.0
4 reviews

9 followers

The lightweight PyTorch wrapper for high-performance AI research. Scale models, not boilerplate. Lightning is one of the most popular deep learning frameworks. Unlike Keras it gives full flexibility. Unlike PyTorch it does not need a ton of boilerplate.