PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.
[](/packages/pytorch-lightning)
<a href="/packages/pytorch-lightning"><img src="/api/badges/pytorch-lightning?period=month" alt="PyPI Stats"></a>