PyTorch agent skills for Claude Code, Codex and other agents.

A skill your agent uses when adding a new PyTorch model to Pyrefly's shape-tracking example corpus under tensor-shapes/pyrefly-torch-stubs/examples — i.e.

facebook/pyrefly

Run Pyrefly benchmarks locally via Buck or Cargo, including PyTorch real-world LSP benchmarks.

facebook/pyrefly

Moves Megatron-LM CI to a newer NVIDIA PyTorch base image, updating both the GitHub and GitLab pins together and handling the CI follow-up.

NVIDIA/Megatron-LM

Port a PyTorch model to use pyrefly's tensor shape type system (Tensor[[B, C, H, W]], Int[T]).

facebook/pyrefly

Check PyTorch ciflow/xpu (xpu.yml) on the main branch, collect the failing XPU test cases from the most recent completed run(s), analyze the ROOT CAUSE of each failure with AI, and produce a list…

intel/torch-xpu-ops

Convert PyTorch ATDISPATCH macros to ATDISPATCHV2 format in ATen C++ code.

intel/torch-xpu-ops

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

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

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

intel/torch-xpu-ops

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

A skill your agent uses when writing or debugging nvMolKit Python code for GPU-accelerated RDKit fingerprints, similarity, conformers, clustering, and molecular searches.

NVIDIA/skills

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

Train ML models on Databricks. An agent skill from databricks/databricks-agent-skills.

databricks/databricks-agent-skills

Official NVIDIA-authored guidance for navigating PhysicsNeMo — pick the model, datapipe, or example for a SciML/AI4Science task (surrogates, forecasting, downscaling, physics-informed, inverse…

NVIDIA/skills

Official NVIDIA-authored guidance for PhysicsNeMo ShardTensor domain parallelism — integrate domain parallelism into training/inference scripts (new or existing) with DDP or FSDP2, write and…

NVIDIA/skills

CLIP vision-language model for image-text retrieval, zero-shot classification, embedding extraction, ONNX export, and TensorRT deployment.

NVIDIA/skills

Fine-tune any HuggingFace CV / VLM / LLM model on local NVIDIA GPUs inside an NGC PyTorch container when no dedicated TAO model skill matches.

NVIDIA/skills

InternVideo2-CLIP L14 (TAO videoclip) for video-text retrieval, zero-shot classification, embedding extraction, LoRA fine-tuning, ONNX export, and TensorRT deployment.

NVIDIA/skills

PyTorch-based TAO image classification. An agent skill from NVIDIA/skills.

NVIDIA/skills

Convert existing PyTorch Lightning training code into an NVFLARE federated job using the Lightning Client API patch, local validation, and job export; use only when the request names…

NVIDIA/skills

Convert existing plain or manual PyTorch training code into an NVFLARE federated job using Client API model exchange, local validation, and job export; use when the user names plain PyTorch or…

NVIDIA/skills

Extract Intel GPU ISA (assembly) from any XPU kernel. An agent skill from intel/torch-xpu-ops.

intel/torch-xpu-ops

A skill your agent uses when asked to implement a fix for a root-caused failure, apply a proposed patch, or produce a staged code change from triage output.

intel/torch-xpu-ops

A skill your agent uses when asked to verify a fix works, confirm a staged patch resolves a failure, or produce a before/after summary of a fix.

intel/torch-xpu-ops

A skill your agent uses when asked to fix a GitHub issue end-to-end, run the agent pipeline on an issue, or process an agent:active / batch tracking issue.

intel/torch-xpu-ops

A skill your agent uses when setting up a new torch-xpu-ops release branch corresponding to a PyTorch release.

intel/torch-xpu-ops

Build PyTorch from source with Intel XPU (GPU) support. An agent skill from intel/torch-xpu-ops.

intel/torch-xpu-ops

Generate PyTorch release notes worksheet for XPU by searching git log between release branches.

intel/torch-xpu-ops

Integrate a HuggingFace Computer Vision model into the NVIDIA TAO Toolkit ecosystem (tao-core config, tao-pytorch trainer, tao-deploy TensorRT pipeline).

NVIDIA/skills

Generate and analyze T1/T2/R roofline reports for PyTorch OOB workloads comparing Intel XPU and NVIDIA CUDA.

intel/torch-xpu-ops

Find upstream PyTorch behavior or fixes that may require XPU parity work, validate them on XPU, and produce independently reviewed evidence.

intel/torch-xpu-ops

A skill your agent uses when asked to reproduce a bug, verify a nightly CI failure, or confirm a failure still exists on latest source.

intel/torch-xpu-ops

Techniques for reducing peak GPU memory in Megatron Bridge — expandable segments, PEFT + SP input re-gather, parallelism resizing, activation recompute, CPU offloading constraints, and common OOM…

NVIDIA/skills

Shallow, text-only triage of a GitHub issue on pytorch or torch-xpu-ops.

intel/torch-xpu-ops

Add unsigned integer (uint) type support to PyTorch operators by updating ATDISPATCH macros.

pytorch/pytorch

Generates music from text descriptions with MusicGen and sound effects with AudioGen, using Meta's AudioCraft PyTorch library with melody and style conditioning.

Orchestra-Research/AI-Research-SKILLs

Write docstrings for PyTorch functions and methods following PyTorch conventions.

pytorch/pytorch

Explains OpenAI's CLIP model for zero-shot image classification, image-text similarity, semantic image search and content moderation, with install steps and code patterns.

Orchestra-Research/AI-Research-SKILLs

Runs an acceptance-style evaluation of an ML GPU cluster: environment dump, matmul FLOPS, all-reduce, disk benchmarks and a dated markdown report.

stas00/ml-engineering

Query PyTorch CI, GitHub Actions, HUD, Grafana, and infrastructure metrics.

pytorch/pytorch

Choose 32-bit vs 64-bit index math in PyTorch CUDA kernels. An agent skill from pytorch/pytorch.

pytorch/pytorch

Measures and shrinks the ExecuTorch runtime binary by building a size test, analyzing it with bloaty and landing each reduction as its own pull request.

pytorch/executorch

Document undocumented public APIs in PyTorch by removing functions from coverageignorefunctions and coverageignoreclasses in docs/source/conf.py, running Sphinx coverage, and adding the appropriate…

pytorch/pytorch

Generate a long-form Chinese interview-prep cheat sheet on a specific ML/LLM topic — formulas with derivations, from-scratch PyTorch code, comparison tables, and 25 高频面试题 (L1 必会 / L2 进阶 / L3 顶级 lab).

wanshuiyin/ARIS-in-AI-Offer

Condensed debugging method and tool recipes for Unix, Python and PyTorch programs: crashes, hangs, segfaults, wrong output, CUDA OOM, NaN values and slowness.

stas00/the-art-of-debugging

Develops production-ready Triton kernels for Liger Kernel. An agent skill from linkedin/Liger-Kernel.

linkedin/Liger-Kernel

Builds ExecuTorch from source: the Python package, C++ runtime, model runners, Android and iOS cross-compilation and backend-specific builds, with environment checks.

pytorch/executorch

Fix bugs reported in PyTorch GitHub issues by reproducing, root-causing, and implementing a fix in the local working tree.

pytorch/pytorch