AI & LLM Engineering · PyTorch

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

Adds PyTorch FSDP2 (fullyshard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing.

Orchestra-Research/AI-Research-SKILLs

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

uw-syfi/vibesys

Build production ML systems with PyTorch 2.x, TensorFlow, and modern ML frameworks.

davila7/claude-code-templates

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

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

alibaba/atrex-kernel-agent

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

Fine-tunes and evaluates OpenVLA-OFT and OFT+ robot policies with LoRA and continuous action heads on LIBERO simulation and ALOHA real-robot setups.

Orchestra-Research/AI-Research-SKILLs

Fine-tunes and serves Physical Intelligence's pi0, pi0-fast and pi0.5 robot policies with JAX or PyTorch, including checkpoint conversion and policy servers.

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

Creates, lists, updates and deletes Volcano Jobs and Queues in KubeSphere, with YAML templates for PyTorch, TensorFlow, MPI and batch jobs plus scheduling troubleshooting.

kubesphere/kubesphere

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

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

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

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

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

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

Sub-skill técnica de Yann LeCun. An agent skill from sickn33/agentic-awesome-skills.

sickn33/agentic-awesome-skills

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