pytorch-lightning
Deep learning framework (PyTorch Lightning / lightning package). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard, MLflow), distributed training (DDP, FSDP, DeepSpeed), for scalable neural network training.
- Version
- 1.1
- License
- Apache-2.0 license
- Compatibility
- Requires Python 3.10+ and lightning 2.6+ (or pytorch-lightning 2.6+). GPU training needs CUDA-capable PyTorch. Optional loggers (wandb, mlflow, comet-ml) and DeepSpeed require separate installs.
Pinned to revision de66e10cd0c8, so it is the text this page describes rather than whatever the author pushed since.
Pre-approved tools experimental
Experimental field. Support varies between clients, so this list is what the author declared, not what your client will enforce.
- Read
- Write
- Edit
- Bash
Files
- skills/pytorch-lightning/SKILL.md
- skills/pytorch-lightning/references/best_practices.md
- skills/pytorch-lightning/references/callbacks.md
- skills/pytorch-lightning/references/data_module.md
- skills/pytorch-lightning/references/distributed_training.md
- skills/pytorch-lightning/references/lightning_module.md
- skills/pytorch-lightning/references/logging.md
- skills/pytorch-lightning/references/trainer.md
- skills/pytorch-lightning/scripts/quick_trainer_setup.py
- skills/pytorch-lightning/scripts/template_datamodule.py
- skills/pytorch-lightning/scripts/template_lightning_module.py
Every link opens the file at its source, pinned to the revision this page describes.