Best agent skills for PyTorch and TensorFlow
We found 131 agent skills for PyTorch and TensorFlow in the catalog; these are the 15 that rank highest. The order comes from our rank, which weighs adoption, freshness, documentation and publisher identity. Nobody can pay for a place on this list. nvflare-convert-pytorch, nvflare-convert-lightning and depth-estimation lead it. pennylane comes from the largest repository here (54,411 stars). 11 of the 15 had activity in the last 30 days. Each entry lists why it made the list and links to its install page.
Updated Oct 26, 2026 · Built from catalog data; how we rank and grade
Top 15 of 131
- 1nvflare-convert-pytorch
NVIDIA FLARE Team <federatedlearning@nvidia.com>
Convert existing plain or manual PyTorch training code into an NVFLARE federated job using Client API model exchange, local validation, and job export; use…
- Repository has 3.7K stars
- Active in the last 30 days
- Open source (Apache-2.0)
★ 3.7KLast active Oct 8, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 2nvflare-convert-lightning
NVIDIA FLARE Team <federatedlearning@nvidia.com>
Convert existing PyTorch Lightning training code into an NVFLARE federated job using the Lightning Client API patch, local validation, and job export; use…
- Repository has 3.7K stars
- Active in the last 30 days
- Open source (Apache-2.0)
★ 3.7KLast active Oct 8, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 3depth-estimation
SharpAI
Real-time depth map privacy transforms using Depth Anything v2 (CoreML + PyTorch)
- Repository has 3.1K stars
- Open source (MIT)
- Ships helper scripts
★ 3.1KLast active Sep 17, 2026Claude Code, Claude apps, Codex, Cursor +6
BFreshness: B–Security: not graded yet - 4model-scaffold
Aperivue
Use when you need a runnable PyTorch training repo for a medical-imaging task (segmentation, classification, detection, synthesis, self-supervised, or…
- Active in the last 30 days
- Open source (MIT)
- Ships helper scripts
★ 349Last active Oct 4, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 5
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 plan…
- Active in the last 30 days
- Open source (MIT)
★ 465Last active Oct 1, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 6pennylane
K-Dense-AI
Builds and differentiates PennyLane quantum circuits, hybrid PyTorch or JAX models, molecular VQE and QAOA workflows.
- Largest repository here (54,411 stars)
- Active in the last 30 days
- Open source (MIT)
★ 54.4KLast active Oct 1, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 7pytorch-lightning
K-Dense-AI
Deep learning framework (PyTorch Lightning / lightning package).
- Largest repository here (54,411 stars)
- Active in the last 30 days
- Open source (MIT)
★ 54.4KLast active Oct 1, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 8
Orchestrates modular PyTorch profiler trace analysis with TraceLens: generates perf reports, prepares category data, runs system-level and compute-kernel…
- Active in the last 30 days
- Open source (MIT)
- Ships helper scripts
★ 465Last active Oct 8, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 9interview-cheatsheet
wanshuiyin
Generate a long-form Chinese interview-prep cheat sheet on a specific ML/LLM topic — formulas with derivations, from-scratch PyTorch code, comparison tables…
- Repository has 18K stars
- Active in the last 30 days
- Open source (MIT)
★ 18KLast active Oct 19, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 10pytorch-fsdp2
Orchestra-Research
Adds PyTorch FSDP2 (fully_shard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing.
- Repository has 13.9K stars
- Open source (MIT)
★ 13.9KLast active Jun 16, 2026Claude Code, Claude apps, Codex, Cursor +6
CFreshness: C–Security: not graded yet - 11pytorch-lightning
Orchestra-Research
High-level PyTorch framework with Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks system, and minimal boilerplate.
- Repository has 13.9K stars
- Open source (MIT)
★ 13.9KLast active Jun 16, 2026Claude Code, Claude apps, Codex, Cursor +6
CFreshness: C–Security: not graded yet - 12mkl-debug-ml-training
00200200
Investigate training failures, NaNs, missing gradients, misleading losses, and non-reproducible runs in PyTorch, Lightning, or TensorFlow/Keras.
- Active in the last 30 days
- Open source (MIT)
★ 48Last active Oct 17, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 13pytorch-patterns
Kilo-Org
PyTorch deep learning patterns and best practices for building robust, efficient, and reproducible training pipelines, model architectures, and data loading.
- Active in the last 30 days
- Open source (Apache-2.0)
★ 195Last active Sep 28, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 14pytorch-lightning
foryourhealth111-pixel
Deep learning framework (PyTorch Lightning).
- Repository has 3.8K stars
- Active in the last 30 days
- Open source (Apache-2.0)
★ 3.8KLast active Oct 18, 2026Claude Code, Claude apps, Codex, Cursor +6
AFreshness: A–Security: not graded yet - 15audiocraft
graniet
PyTorch library for audio generation including text-to-music (MusicGen) and text-to-sound (AudioGen).
- Open source (MIT)
★ 217Last active Jun 27, 2026Claude Code, Claude apps, Codex, Cursor +6
CFreshness: C–Security: not graded yet