Pytorch Lightning
MCP Serverby hxk622 · Added 5mo ago
Claude
4
Install
git clone https://github.com/hxk622/TokenDanceAdd to Claude
claude mcp add pytorch-lightningQuick Setup Config
Copy & paste ready~/Library/Application Support/Claude/claude_desktop_config.json (macOS) %APPDATA%\Claude\claude_desktop_config.json (Windows)
{
"mcpServers": {
"pytorch-lightning": {
"command": "npx",
"args": [
"@modelcontextprotocol/pytorch-lightning"
]
}
}
}Paste into your Claude Desktop config file, then restart Claude.
About
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.
Tags
ai-mlpythongodebuggingdocumentationskill