pytorch-lightning

pytorch-lightning is a skill for Claude Code from dralkh/iktinah. It costs 71 tokens per session (1,713 once invoked), scanned A, a copy of pytorch-lightning, MIT.

A framework that organizes PyTorch code for training neural networks, including data handling, logging, callbacks, and work across multiple processors or accelerators.

In plain words
What is it for?
Use it to structure PyTorch projects, train models on multiple GPUs or TPUs, build data pipelines, and connect training logs to tools such as W&B, TensorBoard, or MLflow.
Why use it?
It removes repeated training setup code and provides a consistent structure for running and tracking larger machine-learning experiments.

Skill for Claude Code

Written for Claude Code: allowed-tools in frontmatter.

Good fit Use it to structure PyTorch projects, train models on multiple GPUs or TPUs, build data pipelines, and connect training logs to tools such as W&B, TensorBoard, or MLflow.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/dralkh/iktinah/pytorch-lightning
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent
npx skills add dralkh/iktinah --skill pytorch-lightning
Clone the repo
git clone --depth 1 https://github.com/dralkh/iktinah

Made for: Claude Code.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for pytorch-lightning

README.md
[![agentmods](https://agentmods.dev/badge/skills/dralkh/iktinah/pytorch-lightning/github.svg)](https://agentmods.dev/skills/dralkh/iktinah/pytorch-lightning)
Your own site
<a href="https://agentmods.dev/skills/dralkh/iktinah/pytorch-lightning"><img src="https://agentmods.dev/badge/skills/dralkh/iktinah/pytorch-lightning/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for pytorch-lightning

Your own site · 80×15
<a href="https://agentmods.dev/skills/dralkh/iktinah/pytorch-lightning"><img src="https://agentmods.dev/badge/skills/dralkh/iktinah/pytorch-lightning.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 71 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 1,713 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. A grade says what 26 rules found in the file — not that it is safe.
Origin 86% copy Near-identical to another mod in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00071 $0.01713
Opus 5 $0.00036 $0.00856
Sonnet 5 $0.00014 $0.00343
Haiku 4.5 $0.00007 $0.00171

Measured 8d ago against content hash e524d755db8f, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-11, from the pricing page.

Security

Grade A, and why

pytorch-lightning scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 8d ago.

The scan reads SKILL.md. This mod also ships 3 executable files (scripts/quick_trainer_setup.py, scripts/template_datamodule.py, scripts/template_lightning_module.py), listed below but not scanned — reading those needs a real analyzer, not pattern matching.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

Origin

This is a copy

86% identical to pytorch-lightning — 20 lines differ, which has more behind it and is treated as the original. This page carries a canonical link to it rather than competing with it.

skills/pytorch-lightning/SKILL.md · 190 lines

How it starts

The opening of the file, as written. The whole thing — 190 lines — stays where its author put it; the contents beside it link to each section on GitHub.

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.

Current upstream: lightning 2.6.4 (PyPI, May 2026). Docs: lightning.ai/docs/pytorch/stable. Use import lightning as L (the pytorch-lightning package name still installs the same library).

Installation

uv pip install lightning

Optional extras:

uv pip install lightning[extra]    # loggers, strategies, etc.
uv pip install wandb mlflow        # specific loggers as needed

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

The Trainer automates the training loop, device management, gradient operations, and callbacks. Key features:

Read the full file on GitHub · 190 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 8d ago First seen · 190 lines · 71 tokens per session scan A e524d755db8f

Subscribe to this mod's changes

pytorch-lightning is a skill published in the GitHub repository dralkh/iktinah (77 stars, last pushed 2mo ago), licensed MIT. It adds 71 tokens to every session and 1,713 once invoked, about $0.0004 per session on Opus 5. A static security scan graded it A with 0 findings. It is 86% identical to pytorch-lightning, differing in 20 lines, and is treated as a copy.