PyTorch · For developers

Deploy AI models to embedded hardware using MathWorks tools (MATLAB, Simulink, Embedded Coder).

matlab/agent-skills-playground

Load this FIRST whenever working with PyTorch CI data (any pytorch/ org repo), the torchci/HUD codebase, or the PyTorch HUD ClickHouse database.

pytorch/test-infra

Publish a Python package to PyPI using uv with credentials loaded from a local .secrets file.

ML4ITS/TimeVQVAE

Expert in PerforatedAI library for adding artificial dendrites to PyTorch neural networks.

PerforatedAI/PerforatedAI

Author a new ShapeShifter graph-transformation pass for AMD Quark (ONNX or PyTorch) so it conforms to the pass framework's conventions and auto-registers.

amd/Quark

End-to-end AscendC custom operator development for Ascend NPU in an ascend-kernel (csrc/ops + build.sh + torchnpu PyTorch custom op) project.

ascend-ai-coding/awesome-ascend-skills

Pick the right ComfyUI startup flags for VRAM, attention, caching, and speed.

artokun/comfyui-mcp

how to use the edit command properly

omegaml/omegaml

Disable or re-enable Cloudflare R2 (download-r2.pytorch.org) usage in managev2.py during R2 outages.

pytorch/test-infra

Profile ExecuTorch model execution. Use when measuring performance, analyzing operator timing, or debugging slow models.

pytorch/executorch

Import PyTorch, ONNX, or Keras 3 / TensorFlow 2.16+ deep learning models into MATLAB as dlnetwork objects.

matlab/matlab-agentic-toolkit

Collect and normalize environment facts (OS, Python, GPU, CUDA/ROCm, container state) before Quark installation or PTQ planning.

amd/Quark

Complete toolkit for Huawei Ascend NPU model conversion and end-to-end inference adaptation.

ascend-ai-coding/awesome-ascend-skills

Guide for writing and modifying NKI kernels. An agent skill from uw-syfi/vibesys.

uw-syfi/vibesys

Benchmarks LLM inference and drives GPU kernel optimization with Magpie.

amd/skills

Review PyTorch upstream unit-test (UT) PRs that enable Intel GPU (XPU) on existing tests.

intel/torch-xpu-ops

Find reverts that landed on pytorch/pytorch main but are missing from a release branch (release/X.Y) because the reverted commit was already shipped in a release candidate, and open one cherry-pick…

pytorch/test-infra

Optimize or review eager CPU-only Albucore PyTorch runtime paths with benchmark-backed decisions.

albumentations-team/albucore

Install or verify the AMD Quark package and its dependencies.

amd/Quark

World-class computer vision skill for image/video processing, object detection, segmentation, and visual AI systems.

davila7/claude-code-templates

World-class ML engineering skill for productionizing ML models, MLOps, and building scalable ML systems.

davila7/claude-code-templates

Shows how to log ML runs, configs, metrics and media with SwanLab and view them in cloud, local or self-hosted dashboards.

Orchestra-Research/AI-Research-SKILLs

Guides technology selection and implementation of AI and ML features in .NET 8+ applications using ML.NET, Microsoft.Extensions.AI (MEAI), Microsoft Agent Framework (MAF), GitHub Copilot SDK, ONNX…

dotnet/skills

Generate (and optionally open) a PyTorch release tracker / cherry-pick tracking issue from a release announcement, like https://github.com/pytorch/pytorch/issues/180506.

pytorch/test-infra

Orchestrates modular PyTorch profiler trace analysis with TraceLens: generates perf reports, prepares category data, runs system-level and compute-kernel subagents in parallel, validates outputs…

amd/skills

Develop and profile the edge-e3 PyTorch-to-NNC flow, including nnc/compiler.py lowering and generated ABI, example/llama smoke models, cpp/libnn runtime operators, BF16 correctness checks, and…

exeex/edge-cores

Build production computer vision pipelines for object detection, tracking, and video analysis.

curiositech/some_claude_skills

Check and compare software component versions on SageMaker HyperPod cluster nodes - NVIDIA drivers, CUDA toolkit, cuDNN, NCCL, EFA, AWS OFI NCCL, GDRCopy, MPI, Neuron SDK (Trainium/Inferentia)…

awslabs/agent-plugins

TRIGGER when user asks to create a microservice that calls Python, runs ML inference, or uses Python libraries (PyTorch, pandas, sentence-transformers, numpy) for its core compute.

microbus-io/fabric

Generate a PyTorch release validation checklist issue by pulling open/closed issues from a GitHub milestone and cherry-picks from a release tracker issue.

pytorch/test-infra

Deep learning framework (PyTorch Lightning / lightning package).

K-Dense-AI/scientific-agent-skills

Multi-GPU setup for PerforatedAI with DataParallel or DistributedDataParallel (DDP).

PerforatedAI/PerforatedAI

Computer vision engineering skill for object detection, image segmentation, and visual AI systems.

alirezarezvani/claude-skills

Run, extend, debug, or review the public edge-e3 bare-metal software harness, including encrypted Verilator builds, hello and tensor examples, all example/llama/model smoke cases, PyTorch BF16…

exeex/edge-cores

Supports PyTorch Geometric (PyG) graph neural networks — node/link/graph classification, message passing (GCN, GAT, GraphSAGE, GIN), heterogeneous graphs, neighbor sampling, and custom datasets.

K-Dense-AI/scientific-agent-skills

Update tools/scripts/generatebinarybuildmatrix.py when a PyTorch release goes live.

pytorch/test-infra

Run a bounded TileLang implementation and optimization task from a PyTorch reference or existing kernel through baseline analysis, a written plan, candidate edits, lowering inspection, validation…

sablin39/tilelang-cuda-skills

Pin (or re-pin) the Linux manywheel builder docker images used by the nightly/release binary build workflows in pytorch/pytorch to a fixed .ci/docker build.

pytorch/test-infra

Set up a working ML training/inference environment on NVIDIA DGX Spark (GB10, aarch64, CUDA 13).

wshobson/agents

Write, review, and run high-level xTBloom Python GFN2-xTB inference with Calculator, Structure, and BatchCalculator, including single systems, repeated geometry updates, heterogeneous ragged…

jinzhezenggroup/computational-chemistry-agent-skills

Root-cause a vLLM torch-nightly CI regression report. An agent skill from pytorch/test-infra.

pytorch/test-infra

Pick Jetson-compatible containers, vLLM runtime images, and Jetson AI Lab PyPI indexes; maps Orin SM 8.7 vs Thor SM 11.0 and JetPack-specific package choices.

NVIDIA/skills

How the on-device narration engine (VieNeu-TTS) and voice cloning work in this system - install tiers, the worker protocol, and the verified traps around torchaudio, voice metadata parsing and…

notivn/AIEV

Universal Runtime best practices for PyTorch inference, Transformers models, and FastAPI serving.

llama-farm/llamafarm

Review vLLM-routed torch-nightly root causes against existing upstream vLLM issues using read-only search results, and emit a validated upstream-checks artifact for the filer.

pytorch/test-infra

Run PyTorch training across GPUs with minimal changes. An agent skill from Luciole-Studio/Misaka-Agent.

Luciole-Studio/Misaka-Agent

Speed up long-sequence transformer training and inference. An agent skill from Luciole-Studio/Misaka-Agent.

Luciole-Studio/Misaka-Agent

Convert existing Hugging Face Transformers Trainer or TRL SFTTrainer training code into an NVFLARE federated job using flare.patch(trainer), local validation, and job export; use when the user names…

NVIDIA/skills