Armory
Source
Browse
Skills

pytorch-lightning-2

Deep learning framework (PyTorch Lightning). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard), distributed training (DDP, FSDP, DeepSpeed), for scalable neural network training.

Score
Unranked
Evidence
No signals yet
Last commit
Not known
Listed

Install

armory install pytorch-lightning-2 --cli claude

writes the skill into.claude/skills/pytorch-lightning-2/SKILL.mdListed as compatible

Configuration
# fetches the source and writes it to:
.claude/skills/pytorch-lightning-2/SKILL.md

Needs the armory CLI · not on npm yet, build it from cli/ in the repository

What it is

Deep learning framework (PyTorch Lightning). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard), distributed training (DDP, FSDP, DeepSpeed), for scalable neural network training.

When to use it

Deep learning framework (PyTorch Lightning). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard), distributed training (DDP, FSDP, DeepSpeed), for scalable neural network training.

How to install / invoke

# Copy the skill into your .claude/skills/ directory
curl -sL https://raw.githubusercontent.com/davila7/claude-code-templates/main/cli-tool/components/skills/scientific/pytorch-lightning/SKILL.md -o .claude/skills/pytorch-lightning/SKILL.md

Notes

Extracted from davila7/claude-code-templates, pytorch-lightning category.