其他

pytorch-lightning

Name: pytorch-lightning
Author: orchestra-research

orchestra-research/ai-research-skills

High-level PyTorch framework with Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks system, and minimal boilerplate. Scales from laptop to supercomputer with same code. Use when you want clean training loops with built-in best practices.

981 次安装
1.4万星标
收录于 2026.01.21

安装命令

$ npx skills add https://github.com/orchestra-research/ai-research-skills --skill pytorch-lightning

SKILL.md

PyTorch Lightning - High-Level Training Framework

Quick start

PyTorch Lightning organizes PyTorch code to eliminate boilerplate while maintaining flexibility.

Installation:

pip install lightning

Convert PyTorch to Lightning (3 steps):

import lightning as L
import torch
from torch import nn
from torch.utils.data import DataLoader, Dataset

SKILL.md

PyTorch Lightning - High-Level Training Framework

Quick start

相关 Skills