pytorch-lightning python library

pytorch-lightning

Subscription: Developer
Last updated an hour ago
Active
Subscription: Developer

A lightweight PyTorch wrapper for machine learning researchers, providing structured training loops, automatic mixed precision, gradient accumulation, and multi-GPU support while reducing boilerplate and improving experiment reproducibility.

pip install pytorch-lightning==2.6.6
  • 2.6.6
    Linux
    Last updated an hour ago