Skill · em Dados, IA e pesquisa

pytorch-lightning

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.

Procedência

Antes de instalar

11 arquivos · 134,7 KB · inclui 3 scripts que executam: scripts/quick_trainer_setup.py, scripts/template_datamodule.py, scripts/template_lightning_module.py

Instalar na sua CLI

O comando baixa a versão fixada (commit 57f899e) direto da origem, para a pasta que a CLI lê. Precisa de curl (macOS e Linux); no Windows não há comando, porque o Rook Labs é para macOS.

Claude Code

Neste projeto: instala em .claude/skills/pytorch-lightning/.

d=".claude/skills/pytorch-lightning"
u="https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/cli-tool/components/skills/scientific/pytorch-lightning"
curl -fsSL --create-dirs \
  -o "$d/SKILL.md" "$u/SKILL.md" \
  -o "$d/references/best_practices.md" "$u/references/best_practices.md" \
  -o "$d/references/callbacks.md" "$u/references/callbacks.md" \
  -o "$d/references/data_module.md" "$u/references/data_module.md" \
  -o "$d/references/distributed_training.md" "$u/references/distributed_training.md" \
  -o "$d/references/lightning_module.md" "$u/references/lightning_module.md" \
  -o "$d/references/logging.md" "$u/references/logging.md" \
  -o "$d/references/trainer.md" "$u/references/trainer.md" \
  -o "$d/scripts/quick_trainer_setup.py" "$u/scripts/quick_trainer_setup.py" \
  -o "$d/scripts/template_datamodule.py" "$u/scripts/template_datamodule.py" \
  -o "$d/scripts/template_lightning_module.py" "$u/scripts/template_lightning_module.py" \
  -o "$d/LICENSE" "https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/LICENSE"

Global: instala em ~/.claude/skills/pytorch-lightning/.

d="$HOME/.claude/skills/pytorch-lightning"
u="https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/cli-tool/components/skills/scientific/pytorch-lightning"
curl -fsSL --create-dirs \
  -o "$d/SKILL.md" "$u/SKILL.md" \
  -o "$d/references/best_practices.md" "$u/references/best_practices.md" \
  -o "$d/references/callbacks.md" "$u/references/callbacks.md" \
  -o "$d/references/data_module.md" "$u/references/data_module.md" \
  -o "$d/references/distributed_training.md" "$u/references/distributed_training.md" \
  -o "$d/references/lightning_module.md" "$u/references/lightning_module.md" \
  -o "$d/references/logging.md" "$u/references/logging.md" \
  -o "$d/references/trainer.md" "$u/references/trainer.md" \
  -o "$d/scripts/quick_trainer_setup.py" "$u/scripts/quick_trainer_setup.py" \
  -o "$d/scripts/template_datamodule.py" "$u/scripts/template_datamodule.py" \
  -o "$d/scripts/template_lightning_module.py" "$u/scripts/template_lightning_module.py" \
  -o "$d/LICENSE" "https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/LICENSE"

Codex

Neste projeto: instala em .agents/skills/pytorch-lightning/.

d=".agents/skills/pytorch-lightning"
u="https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/cli-tool/components/skills/scientific/pytorch-lightning"
curl -fsSL --create-dirs \
  -o "$d/SKILL.md" "$u/SKILL.md" \
  -o "$d/references/best_practices.md" "$u/references/best_practices.md" \
  -o "$d/references/callbacks.md" "$u/references/callbacks.md" \
  -o "$d/references/data_module.md" "$u/references/data_module.md" \
  -o "$d/references/distributed_training.md" "$u/references/distributed_training.md" \
  -o "$d/references/lightning_module.md" "$u/references/lightning_module.md" \
  -o "$d/references/logging.md" "$u/references/logging.md" \
  -o "$d/references/trainer.md" "$u/references/trainer.md" \
  -o "$d/scripts/quick_trainer_setup.py" "$u/scripts/quick_trainer_setup.py" \
  -o "$d/scripts/template_datamodule.py" "$u/scripts/template_datamodule.py" \
  -o "$d/scripts/template_lightning_module.py" "$u/scripts/template_lightning_module.py" \
  -o "$d/LICENSE" "https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/LICENSE"

Global: instala em ~/.agents/skills/pytorch-lightning/.

d="$HOME/.agents/skills/pytorch-lightning"
u="https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/cli-tool/components/skills/scientific/pytorch-lightning"
curl -fsSL --create-dirs \
  -o "$d/SKILL.md" "$u/SKILL.md" \
  -o "$d/references/best_practices.md" "$u/references/best_practices.md" \
  -o "$d/references/callbacks.md" "$u/references/callbacks.md" \
  -o "$d/references/data_module.md" "$u/references/data_module.md" \
  -o "$d/references/distributed_training.md" "$u/references/distributed_training.md" \
  -o "$d/references/lightning_module.md" "$u/references/lightning_module.md" \
  -o "$d/references/logging.md" "$u/references/logging.md" \
  -o "$d/references/trainer.md" "$u/references/trainer.md" \
  -o "$d/scripts/quick_trainer_setup.py" "$u/scripts/quick_trainer_setup.py" \
  -o "$d/scripts/template_datamodule.py" "$u/scripts/template_datamodule.py" \
  -o "$d/scripts/template_lightning_module.py" "$u/scripts/template_lightning_module.py" \
  -o "$d/LICENSE" "https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/LICENSE"

Antigravity

Neste projeto: instala em .agents/skills/pytorch-lightning/.

d=".agents/skills/pytorch-lightning"
u="https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/cli-tool/components/skills/scientific/pytorch-lightning"
curl -fsSL --create-dirs \
  -o "$d/SKILL.md" "$u/SKILL.md" \
  -o "$d/references/best_practices.md" "$u/references/best_practices.md" \
  -o "$d/references/callbacks.md" "$u/references/callbacks.md" \
  -o "$d/references/data_module.md" "$u/references/data_module.md" \
  -o "$d/references/distributed_training.md" "$u/references/distributed_training.md" \
  -o "$d/references/lightning_module.md" "$u/references/lightning_module.md" \
  -o "$d/references/logging.md" "$u/references/logging.md" \
  -o "$d/references/trainer.md" "$u/references/trainer.md" \
  -o "$d/scripts/quick_trainer_setup.py" "$u/scripts/quick_trainer_setup.py" \
  -o "$d/scripts/template_datamodule.py" "$u/scripts/template_datamodule.py" \
  -o "$d/scripts/template_lightning_module.py" "$u/scripts/template_lightning_module.py" \
  -o "$d/LICENSE" "https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/LICENSE"

Global: instala em ~/.gemini/antigravity-cli/skills/pytorch-lightning/.

d="$HOME/.gemini/antigravity-cli/skills/pytorch-lightning"
u="https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/cli-tool/components/skills/scientific/pytorch-lightning"
curl -fsSL --create-dirs \
  -o "$d/SKILL.md" "$u/SKILL.md" \
  -o "$d/references/best_practices.md" "$u/references/best_practices.md" \
  -o "$d/references/callbacks.md" "$u/references/callbacks.md" \
  -o "$d/references/data_module.md" "$u/references/data_module.md" \
  -o "$d/references/distributed_training.md" "$u/references/distributed_training.md" \
  -o "$d/references/lightning_module.md" "$u/references/lightning_module.md" \
  -o "$d/references/logging.md" "$u/references/logging.md" \
  -o "$d/references/trainer.md" "$u/references/trainer.md" \
  -o "$d/scripts/quick_trainer_setup.py" "$u/scripts/quick_trainer_setup.py" \
  -o "$d/scripts/template_datamodule.py" "$u/scripts/template_datamodule.py" \
  -o "$d/scripts/template_lightning_module.py" "$u/scripts/template_lightning_module.py" \
  -o "$d/LICENSE" "https://raw.githubusercontent.com/davila7/claude-code-templates/57f899e5394bb8ca166f38eacae8f0853cbfe033/LICENSE"

Peça ao Rook

Já usa o Rook Labs? Cole no chat do Rook: instale a skill https://rooklabs.sh/marketplace/cct.pytorch-lightning.scientific

Prévia do SKILL.md

---
name: pytorch-lightning
description: "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."
---

# PyTorch Lightning

## Overview

PyTorch Lightning is a deep learning framework that organizes PyTorch code to eliminate boilerplate while maintaining full flexibility. Automate training workflows, multi-device orchestration, and implement best practices for neural network training and scaling across multiple GPUs/TPUs.

## When to Use This Skill

This skill should be used when:
- Building, training, or deploying neural networks using PyTorch Lightning
- Organizing PyTorch code into LightningModules
- Configuring Trainers for multi-GPU/TPU training
- Implementing data pipelines with LightningDataModules
- Working with callbacks, logging, and distributed training strategies (DDP, FSDP, DeepSpeed)
- Structuring deep learning projects professionally

## Core Capabilities

### 1. LightningModule - Model Definition

Organize PyTorch models into six logical sections:

1. **Initialization** - `__init__()` and `setup()`
2. **Training Loop** - `training_step(batch, batch_idx)`
3. **Validation Loop** - `validation_step(batch, batch_idx)`
4. **Test Loop** - `test_step(batch, batch_idx)`
5. **Prediction** - `predict_step(batch, batch_idx)`
6. **Optimizer Configuration** - `configure_optimizers()`

**Quick template reference:** See `scripts/template_lightning_module.py` for a complete boilerplate.

**Detailed documentation:** Read `references/lightning_module.md` for comprehensive method documentation, hooks, properties, and best practices.

### 2. Trainer - Training Automation
…

Ver todo o marketplace