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.
Details
- Path
- skills/pytorch-lightning
- License
- Apache-2.0 license
- Allowed tools
- 1
- Bundled scripts
- 3
- Dependencies
- 3
Allowed tools
Read Write Edit Bash
Bundled scripts
- skills/pytorch-lightning/scripts/template_datamodule.py
- skills/pytorch-lightning/scripts/quick_trainer_setup.py
- skills/pytorch-lightning/scripts/template_lightning_module.py