Skip to main content
affpapa.org
RU EN
skill

pytorch-lightning · davila7 — Claude Code skill: what it does, how to install

Helps Claude Code structure PyTorch code into LightningModules and configure multi-GPU or TPU training. It removes boilerplate loops and organizes data pipelines.

Install into Claude Code · claude-code-templates All methods and platforms ↓ 
npx claude-code-templates@latest --skill scientific/pytorch-lightning
Open on GitHub

What pytorch-lightning does

This skill guides Claude Code in refactoring raw PyTorch code into structured Lightning projects. Model logic resides inside LightningModule, while dataset preparation lives in LightningDataModule. The skill provides starter scripts like template_lightning_module.py and template_datamodule.py.

For model execution, the skill guides Trainer configuration. It sets up mixed precision and gradient clipping. Metric tracking connects directly to WandbLogger or TensorBoard. Built-in hooks include EarlyStopping and ModelCheckpoint. Running fast_dev_run=True tests pipeline integrity on a single batch.

Device distribution rules depend on weight count: DDP handles networks below 500M parameters, while FSDP or DeepSpeed fit larger models. The resulting code will fail without torch and lightning installed in your environment.

What it helps with

  • convert raw PyTorch training scripts into a modular LightningModule
  • isolate dataset preparation and data loaders inside a LightningDataModule
  • configure multi-GPU training with DDP, FSDP, or DeepSpeed
  • add automated checkpointing and early stopping callbacks

Example prompts for the agent

  • Convert the PyTorch network in model.py into a LightningModule with Adam optimizer
  • Write a LightningDataModule that loads training and validation datasets with batch size 32
  • Configure a Trainer for 4 GPUs using FSDP distributed strategy

How to install pytorch-lightning

1claude-code-templates

npx claude-code-templates@latest --skill scientific/pytorch-lightning

2Skills CLI (Vercel)

npx skills add davila7/claude-code-templates --skill pytorch-lightning -a claude-code

3Manually: put the files in place

~/.claude/skills/pytorch-lightning/SKILL.md

Copy the whole skill folder; for one project use .claude/skills/pytorch-lightning/

Step by step, with commands for each platform: How to install skills in Claude Code: five ways

Questions about pytorch-lightning

How to install pytorch-lightning in Claude Code?

Run npx claude-code-templates@latest --skill scientific/pytorch-lightning. Alternatively, install it with npx skills add davila7/claude-code-templates --skill pytorch-lightning -a claude-code.

What does the pytorch-lightning skill do?

It provides templates and guidelines for Claude Code to structure deep learning projects. The agent learns how to separate model steps, build data pipelines, and manage training routines.

Which distributed strategy does the skill recommend for training?

The documentation recommends DDP for models with fewer than 500M parameters. For architectures with 500M or more parameters, it directs users toward FSDP or DeepSpeed.

Similar skills

All skills for AI and ML →

pytorch-lightning by other authors comes first.

pytorch-lightning K-Dense-AI Guides agents through structuring PyTorch Lightning 2.6.6 code, from LightningModules to distributed trainers. Useful when scaling training across GPUs. CX Codex +1 2,000installs skill-creator official anthropics Creates and refines skills, then tests them with prompts, evaluations, and performance checks. 414,180installs writing-for-agents Matt Pocock Guidance for writing skills and agent-facing documents. Use it when editing AGENTS.md, CLAUDE.md, or reference files reached from them. 466,719installs media-use heygen-com Finds or generates media for HyperFrames videos, including voiceovers, music, cutouts, and color grades. Stores files locally in assets. CX Codex +1 1,100,000installs mcp-builder official anthropics This skill walks through building an MCP server in TypeScript or Python, from reading the protocol spec to writing tests and evaluation questions. 130,949installs subagent-driven-development Jesse Vincent This skill runs an implementation plan through fresh subagents for independent tasks and reviews each result. Use it for multi-step work in one session. CX Codex 237,746installs orchestration stablyai Coordinates supervised Orca workers with task DAGs, decision gates, and threaded messages. Fits workflows that split work across multiple agents. CX Codex +1 350,300installs

From the same set804

The whole set →
command-creator Assists in designing reusable Claude Code slash commands. Sets up parameters, picks storage scope, and generates structured Markdown files for agents. 54installs plan-mode-game This hook opens a browser game when Claude Code enters Plan Mode and tracks tool activity. It gives you something to do while a plan takes shape. 54installs szamlazz Issue, cancel, and download Hungarian invoices through the szamlazz.hu Agent API. It also looks up taxpayer details and keeps a reusable customer cache. 54installs scholar-evaluation Assess papers, proposals, and literature reviews against research criteria. Use it to organize feedback on methods, evidence, analysis, and academic writing. 53installs naming-analyzer Reviews variable, function, class and file names for clarity, spots misleading labels and offers alternatives that follow language conventions. 53installs kotlin-coroutines-expert This skill guides Claude Code through Kotlin Coroutines and Flow, from coordinating concurrent work to handling failures and testing asynchronous functions. Use it when debugging cancellation or building reactive data streams. 53installs aeon Gives Claude a working map of the aeon Python library for classification, forecasting, clustering and anomaly detection on time series. 52installs resume-tailor This skill reshapes a resume for a particular job while keeping the candidate’s claims accurate. Use it when one base resume needs a role-specific version. 52installs

Project sponsors

Companies that keep this analytics open
Gold sponsor
Nutra.Club

Impressions, clicks and code copies — open statistics. We keep the numbers public for the same reason NeBlask itself is public. Become a sponsor