PyTorch · For data scientists

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

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

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

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

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

RightNow-AI/openfang

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

affaan-m/ECC

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

ZJLi2013/awesome-kernel-skills

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

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

sickn33/agentic-awesome-skills

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

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

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 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

Route Quark user goals to the correct atomic skill or workflow.

amd/Quark

Route Quark user goals to the correct atomic skill or workflow.

amd/Quark

Validate Triton-Ascend kernel outputs against PyTorch references with dtype-aware tolerances, exact integer checks, bfloat16 promotion, and boolean handling.

Krusty84/triton-ascend-agent-dev-kit

This skill provides guidance for implementing PyTorch pipeline parallelism for distributed training of large language models.

lazyFrogLOL/Harness_Engineering

This skill provides guidance for implementing tensor parallelism in PyTorch, specifically column-parallel and row-parallel linear layers.

lazyFrogLOL/Harness_Engineering

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

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

intel/torch-xpu-ops

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

jinzhezenggroup/computational-chemistry-agent-skills

Design and execute computational social science analyses across 11 modules: text-as-data/NLP (STM, BERTopic, Wordfish, BERT, conText embedding regression, LLM annotation + DSL bias correction…

joshzyj/open-scholar-skill

Classical ML in Python: classification, regression, clustering, dim reduction, evaluation, tuning, preprocessing pipelines.

jaechang-hits/SciAgent-Skills

通过 PyTorch torch.distributed 接口测试昇腾 NPU 通信算子性能。支持指定任意 tensor shape、dtype，使用 torchrun 启动，贴近真实训练场景的通信算子测试与性能分析。Use for testing collective communication operators (AllReduce, AllGather, ReduceScatter…

ascend-ai-coding/awesome-ascend-skills

Build pytorch model trainer operations. An agent skill from jeremylongshore/tons-of-skills-marketplace.

jeremylongshore/tons-of-skills-marketplace

Run distributed GPU training jobs on CoreWeave with multi-node PyTorch.

jeremylongshore/tons-of-skills-marketplace

Avoid common PyTorch mistakes and apply robust training patterns

wentorai/research-plugins

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

secondsky/claude-skills

GPU optimization for consumer NVIDIA GPUs (8-24GB VRAM) covering mixed precision, gradient checkpointing, XGBoost GPU, CuPy/cuDF migration, and torch.compile.

Mathews-Tom/armory

Use this sub-skill when constructing torchvision datasets, validating dataset roots and downloads, wrapping datasets for transforms v2, decoding or encoding images with torchvision.io, or using…

VectorSpaceLab/AREX-Skill

A skill your agent uses when running RobustVideoMatting inference, loading weights, using convertvideo or TorchHub, converting image/video inputs, or reasoning about exported model runtimes.

VectorSpaceLab/AREX-Skill

A skill your agent uses when working with RobustVideoMatting MattingNetwork APIs, recurrent states, model variants, refiners, tensor shapes, and safe synthetic forward checks.

VectorSpaceLab/AREX-Skill

Use torchsummary.summary and torchsummary.summarystring to inspect PyTorch nn.Module shapes, parameter counts, devices, dtypes, and memory estimates.

VectorSpaceLab/AREX-Skill

Use torchvision.ops for boxes, NMS, ROI pooling/alignment, detection helper concepts, and extension-dependent troubleshooting.

VectorSpaceLab/AREX-Skill

A skill your agent uses for PaddleViT's DINO self-supervised vision-transformer pretraining, multi-crop data contracts, teacher/student configuration, single- or multi-GPU launch planning…

VectorSpaceLab/AREX-Skill

Use Torch Points3D for point-cloud deep learning APIs, datasets/transforms, Hydra training/evaluation, checkpoints, sparse backend decisions, and registration workflows.

VectorSpaceLab/AREX-Skill

Use TorchMetrics to choose, inspect, and combine metric families for PyTorch evaluation, including core API, domain metrics, model-based metrics, and wrappers.

VectorSpaceLab/AREX-Skill