13 tools that work with PyTorch: 2 MCP servers, 11 Agent Skills and 0 Claude Code plugins. 0 of the remote servers answered our live check, 0 worked without an API key, and 0 come from an official or verified publisher. Ranked by GitHub stars plus official, live and free-to-try signals.
Most common uses: Code search, analysis, refactoring & languages (6), Models, LLM APIs, routing, serving, training & GPUs (5), Notes, docs & knowledge bases (1), Science: biology, chemistry, physics & engineering (1).
| Tool | Stars | Last update | License | |
|---|---|---|---|---|
| distributed-training-accelerate Simplest distributed training API. 4 lines to add distributed support to any PyTorch script. Unified API for DeepSpeed/FSDP/Megatron/DDP. Automatic de SkillInstall · skills CLI | 33,495 | MIT | ||
| distributed-training-pytorch-lightning High-level PyTorch framework with Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks system, and minimal boilerplate. Scale SkillInstall · skills CLI | 33,495 | MIT | ||
| interview-cheatsheet Generate a long-form Chinese interview-prep cheat sheet on a specific ML/LLM topic — formulas with derivations, from-scratch PyTorch code, comparison SkillInstall · skills CLI | 19,162 | MIT | ||
| ml-training-recipes Battle-tested PyTorch training recipes for all domains — LLMs, vision, diffusion, medical imaging, protein/drug discovery, spatial omics, genomics. Co SkillInstall · skills CLI | 14,526 | MIT | ||
| pytorch-fsdp2 Adds PyTorch FSDP2 (fully_shard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing. Use w SkillInstall · skills CLI | 14,526 | MIT | ||
| pytorch-lightning Deep learning framework (PyTorch Lightning). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelin SkillInstall · skills CLI | 3,816 | Apache-2.0 | ||
| torchdrug Graph-based drug discovery toolkit. Molecular property prediction (ADMET), protein modeling, knowledge graph reasoning, molecular generation, retrosyn SkillInstall · skills CLI | 3,816 | Apache-2.0 | ||
| quark-install Installs or verifies AMD Quark and ensures the selected Python environment has an accelerator-matched PyTorch. Applies to "install Quark", "set up Qua SkillInstall · skills CLI | 490 | MIT | ||
| quark-torch-llm-ptq Runs an end-to-end AMD Quark post-training quantization workflow for PyTorch / Hugging Face LLMs: inspect a Hub or local model, choose a quantization SkillInstall · skills CLI | 490 | MIT | ||
| VisualTorch Render PyTorch architecture diagrams and animated GIF reveals from trusted model source. MCP serverGitHub-verified authorLocal · PyPI | 350 | MIT | ||
| testing-fwd-bwd-kernels Implement and validate forward and backward TileLang kernels integrated with torch.autograd.Function. Use for custom differentiable operators and grad SkillInstall · skills CLI | 149 | — | ||
| torch-profiling-tilelang-programs Inspect a PyTorch baseline with torch.compile graphs/code and torch.profiler before drafting a custom kernel, then compare TileLang and PyTorch timeli SkillInstall · skills CLI | 149 | — | ||
| io.github.RightNow-AI/forge-mcp-server Turn PyTorch into fast CUDA/Triton kernels on real datacenter GPUs with up to 14x speedup. MCP serverGitHub-verified authorLocal · npm | — | — |