PyTorch agent skills, page 4

Benchmarks LLM inference and drives GPU kernel optimization with Magpie.

amd/skills

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

Learn the target framework from enabled knowledge tools and implement a baseline GPU kernel.

alibaba/atrex-kernel-agent

Set up conditions for PINA problems. An agent skill from PINA-org/PINA.

PINA-org/PINA

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

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

alirezarezvani/claude-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

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

Builds and differentiates PennyLane quantum circuits, hybrid PyTorch or JAX models, molecular VQE and QAOA workflows.

K-Dense-AI/scientific-agent-skills

Trains and evaluates single-agent reinforcement learning with Stable Baselines3 (PPO, SAC, DQN, TD3, DDPG, A2C), Gymnasium custom environments, vectorized rollouts, callbacks, and checkpoint…

K-Dense-AI/scientific-agent-skills

Builds and troubleshoots TorchDrug 0.2.1 workflows for molecular graphs, property prediction, self-supervised pretraining, molecule generation, retrosynthesis, protein representation learning, and…

K-Dense-AI/scientific-agent-skills

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

PerforatedAI/PerforatedAI

Write code that calls the installed nvMolKit Python API for GPU-accelerated, batched RDKit-style operations - Morgan fingerprints, Tanimoto/cosine similarity, ETKDG conformer embedding, MMFF/UFF…

NVIDIA-BioNeMo/bionemo-agent-toolkit

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

Machine learning engineer expert for PyTorch, scikit-learn, model evaluation, and MLOps

RightNow-AI/openfang

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

pytorch/test-infra

PyTorch深度学习模式与最佳实践，用于构建稳健、高效且可复现的训练流程、模型架构和数据加载. An agent skill from affaan-m/ECC.

affaan-m/ECC

Featurizes small molecules with Molfeat for QSAR/QSPR, chemical similarity, virtual screening, and molecular ML.

K-Dense-AI/scientific-agent-skills

Unified kernel benchmarking protocol producing JSON results with latency, TFLOPS, GBps, and comparison against PyTorch baselines.

ZJLi2013/awesome-kernel-skills

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

Multi-GPU and distributed training patterns with PyTorch DDP.

aiming-lab/AutoResearchClaw

Best practices for building robust PyTorch training loops. An agent skill from aiming-lab/AutoResearchClaw.

aiming-lab/AutoResearchClaw

Fine-tune a DPA3 model in DeePMD-kit using the PyTorch backend.

jinzhezenggroup/computational-chemistry-agent-skills

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

wshobson/agents

A skill your agent uses when creating, migrating, or debugging pixi environments, especially for scientific Python, bioinformatics, single-cell analysis, CUDA/PyTorch, Jupyter/VS Code kernels…

xuzhougeng/wisp-science

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

A skill your agent uses when you need a runnable PyTorch training repo for a medical-imaging task (segmentation, classification, detection, synthesis, self-supervised, or fine-tuning a pretrained…

Aperivue/medsci-skills

Automatically discover machine learning and AI skills when working with machine learning, PyTorch, training, inference, RAG, embeddings, fine-tuning, LLM, DSPy, HuggingFace, or diffusion models.

rand/cc-polymath

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

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

Build and debug masked one-dimensional elementwise kernels for Triton-Ascend, including launch wrappers and PyTorch/NPU correctness checks.

Krusty84/triton-ascend-agent-dev-kit

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

Fully sharded data-parallel training for large models. An agent skill from Luciole-Studio/Misaka-Agent.

Luciole-Studio/Misaka-Agent

Pretrain LLMs at scale with PyTorch 4D parallelism. An agent skill from Luciole-Studio/Misaka-Agent.

Luciole-Studio/Misaka-Agent

Fallback loader for the canonical PyTorch ci-metrics skill. An agent skill from pytorch/test-infra.

pytorch/test-infra

Train ML models with scikit-learn, PyTorch, TensorFlow. An agent skill from secondsky/claude-skills.

secondsky/claude-skills

Sub-skill técnica de Yann LeCun. An agent skill from majiayu000/claude-skill-registry.

majiayu000/claude-skill-registry

Apply existing ShapeShifter graph passes to an .onnx model via the quark-cli shapeshifter CLI or a ShapeShifter YAML.

amd/Quark

Diagnose failed Quark installation, PTQ execution, script generation, or export attempts.

amd/Quark

Install or verify the correct PyTorch build for a user's accelerator backend before Quark installation.

amd/Quark