pytorch-lightning

Guides the creation of structured PyTorch training code with reusable data pipelines and scalable training configurations.

When to use it

Use when building neural networks with PyTorch Lightning or configuring their data pipelines, callbacks, logging, and distributed training.

It changes how the agent organizes and implements PyTorch model definitions, training loops, evaluation hooks, and optimizer configuration.