PyTorch Lightning
· Batch Import
Description
A high-level training framework for PyTorch that automates 40+ engineering details like epoch loops, optimization, and hardware acceleration while maintaining flexibility. Supports multi-GPU/multi-node scaling, reproducibility, and decouples research code from engineering boilerplate.
Repository
https://github.com/majiayu000/claude-skill-registry/tree/main/skills/development/pytorch-lightning
View on GitHub