PyTorch

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

intel/torch-xpu-ops

Installs or verifies AMD Quark and ensures the selected Python environment has an accelerator-matched PyTorch.

amd/Quark

Runs an end-to-end AMD Quark post-training quantization workflow for PyTorch / Hugging Face LLMs: inspect a Hub or local model, choose a quantization plan, create reproducible artifacts, request…

amd/Quark

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

jinzhezenggroup/computational-chemistry-agent-skills

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

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 Geometric (PyG) for graph neural networks: node/graph classification, link prediction with GCN, GAT, GraphSAGE, GIN.

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

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

优化实际模型推理链路，将正确性对齐、分段 profiling、显存与数据搬运、TensorRT/ONNX/PyTorch 后端、attention/kernel、FP8/compile、缓存与少步采样、质量回归、GPU 成本和服务验收串成同一实验闭环。当用户要求推理提速、降低显存或 GPU 成本、复现模型效果、定位 GPU 利用率低、优化图像/视频/扩散模型或自托管 LLM 时使用，提供…

majiayu000/spellbook

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

PyTorch-native Graph Neural Network framework for molecules and proteins.

aipoch/medical-research-skills

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

intel/torch-xpu-ops

Optimize transformer attention with Flash Attention — 2-4x speedup, 10-20x memory reduction for long sequences on CUDA GPUs.

AlexAI-MCP/hermes-CCC

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

Build a ChatGPT-like LLM from scratch using PyTorch step by step

wentorai/research-plugins

Avoid common PyTorch mistakes and apply robust training patterns

wentorai/research-plugins

PyTorch Lightning framework for scalable model training and research

wentorai/research-plugins

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

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

A skill your agent uses when selecting Sentence Transformers inference backends or exporting/optimizing models for PyTorch, ONNX, or OpenVINO.

VectorSpaceLab/AREX-Skill

Use this sub-skill for Torch-TensorRT model compilation, dynamic input planning, torch.export workflows, save/load formats, raw TensorRT engines, and compile-time troubleshooting.

VectorSpaceLab/AREX-Skill

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

A skill your agent uses for Torch-TensorRT tasks: compiling PyTorch models with TensorRT, dynamic-shape/export workflows, runtime optimization, Triton/C++/distributed deployment, debugging…

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

A skill your agent uses when working with TorchVision models, weights, transforms, TVTensors, datasets, image IO, visualization utilities, vision ops, detection helpers, or official reference…

VectorSpaceLab/AREX-Skill

Use Towhee's optional PyTorch Trainer, TrainingConfig YAML, NNOperator training bridge, and towhee.models package boundaries safely.

VectorSpaceLab/AREX-Skill

A skill your agent uses when planning or auditing TorchVision reference training/evaluation workflows for classification, quantization, detection, segmentation, video classification, optical flow…

VectorSpaceLab/AREX-Skill

Guide safe video-file, webcam, and optional half-precision demo use for pytorch-yolo-v3.

VectorSpaceLab/AREX-Skill

A skill your agent uses for clone-run Ultralytics YOLOv5 workflows: detection, segmentation, classification, export, benchmarks, datasets, weights, and Flask REST serving.

VectorSpaceLab/AREX-Skill

Ankh 蛋白质语言模型昇腾 NPU 迁移 Skill，适用于 Ankh base/large、Ankh3 large/XL 以及同类基于 HuggingFace Transformers 与 PyTorch 的蛋白模型从 CUDA/GPU 到华为 Ascend NPU 的环境检查、代码适配、权重加载、验证脚本补齐与文档沉淀。

ascend-ai-coding/awesome-ascend-skills

昇腾 TensorFlow Community 迁移适配 Skill，适用于将基于 TensorFlow 2.x 的模型原生部署到华为 Ascend NPU，而不经过 TF 到 PyTorch 转换，覆盖 aarch64 源码编译 TF 2.6.5、tfplugin 安装、自动迁移工具使用、手动适配与精度验证。

ascend-ai-coding/awesome-ascend-skills

DeepFRI 的 TensorFlow 到 PyTorch 转换与昇腾 NPU 迁移 Skill，适用于蛋白质功能预测场景下的 TF 模型分析、PyTorch 重写、权重逐层映射、NPU 推理与精度验证，尤其适合需要在 Ascend 上运行 DeepFRI CNN 或 GCN 路径时使用。

ascend-ai-coding/awesome-ascend-skills

DeepFRI TensorFlow 原生昇腾 NPU 迁移 Skill，适用于不做 TF 到 PyTorch 转换、而是直接使用 TensorFlow 2.6.5 与 npudevice 在华为 Ascend 上运行 DeepFRI 的场景，覆盖源码编译、tfplugin 安装、代码适配、推理与 CPU 对比验证。

ascend-ai-coding/awesome-ascend-skills

OligoFormer 昇腾 NPU 迁移 Skill，适用于将基于 PyTorch Transformer 的 siRNA 效能预测模型迁移到华为 Ascend NPU，覆盖环境搭建、RNA-FM 依赖安装、代码适配、推理验证与可选训练流程。

ascend-ai-coding/awesome-ascend-skills

ProteinBERT 昇腾 NPU 部署与迁移 Skill，适用于将 TensorFlow 或 Keras 版 ProteinBERT 转成基于 PyTorch 与 torchnpu 的实现，覆盖权重转换、embedding 提取、微调训练、注意力可视化和 GPU 与 NPU 精度验证。

ascend-ai-coding/awesome-ascend-skills

TensorFlow 或 Keras 模型改写到 PyTorch 的通用 Skill，适用于在华为 Ascend NPU 或其他依赖 PyTorch 生态的平台上完成层级映射、权重转换、逐层数值验证和端到端精度对比，尤其适合 ProteinBERT、DeepFRI 这类科学模型的跨框架迁移。

ascend-ai-coding/awesome-ascend-skills