AI & LLM Engineering · PyTorch

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

HuggingFace Diffusers 环境配置指南，用于华为昇腾 NPU。覆盖 CANN 版本检测、PyTorch + torchnpu 安装、Diffusers 库安装及环境验证。当用户需要在昇腾 NPU 上配置 Diffusers 环境时使用。

ascend-ai-coding/awesome-ascend-skills

基于 PyTorch 框架的昇腾 NPU 模型推理融合算子优化技能。分析模型代码，识别可替换为 torchnpu 融合算子的计算模式，生成替换方案。触发场景：torchnpu 融合算子替换、MoE/Attention/FFN/Norm 等模块的推理算子适配、torchnpu API 使用咨询。基于仓库已有模型的融合算子经验，按计算语义推荐最佳方案。

ascend-ai-coding/awesome-ascend-skills

基于 PyTorch 框架的昇腾 NPU 模型推理适配与部署基线技能。从 HF 链接或本地模型代码出发，按 cann-recipes-infer 仓库规范适配到 ModelRunner 推理框架，输出可运行的标准模型目录和性能基线数据。触发场景：新模型适配到昇腾 NPU 推理框架、已有模型的部署基线采集、模型迁移和初始跑通验证。

ascend-ai-coding/awesome-ascend-skills

从用户 PyTorch/Python 代码中提取算子实现，构建为算子任务格式的标准化 任务文件。支持两种模式：单 case（单一自包含 .py，getinputs 返回单组）和 多 case（.py + 同名 .json 配对，getinputgroups 返回多组）。

ascend-ai-coding/awesome-ascend-skills

根据设计文档生成 AscendC 算子完整代码实现并完成框架适配。TRIGGER when: 设计文档已完成，需要生成 ophost/opkernel 代码、注册到 PyTorch 框架、编译测试。关键词：代码生成、ophost、opkernel、tiling、kernel、框架适配、算子注册。

ascend-ai-coding/awesome-ascend-skills

面向 Ascend PyTorch Profiler / msprof DB（如 ascendpytorchprofiler.db、msprof.db）的 SQL 分析技能。将自然语言问题（算子耗时、通信、下发、调度、schema/table 查询）转为安全可执行 SQL，并按需从官方文档提取表结构详情。

ascend-ai-coding/awesome-ascend-skills

MindSpeed-MM multimodal model suite environment setup guide for Huawei Ascend NPU.

ascend-ai-coding/awesome-ascend-skills

A skill your agent uses when training or debugging a neural net in PyTorch — the forward/loss/backward/step loop and its silent bugs, mixed precision (AMP), AdamW/LR schedules, DDP/FSDP/ZeRO…

ericrisco/rsc-harness

AI 模型文件逆向与静态分析：ONNX/PyTorch/Safetensors/TFLite 格式解析、 网络结构还原、权重提取、文件级水印分析（权重 pattern/metadata/tensor hash/embedding 异常）。

dslsdzc/rev-skills

Triton language skill for Python GPU kernel authoring. An agent skill from mohitmishra786/low-level-dev-skills.

mohitmishra786/low-level-dev-skills

Guide for building Graph Neural Networks with PyTorch Geometric (PyG).

BioTender-max/awesome-bio-agent-skills

使用本地 BiRefNet GGUF 模型完成图片或视频抠图、人物抠图、主体分割和背景移除，并输出透明 PNG、MOV 或 WebM。适用于用户提到图片抠图、照片去背景、人像透明图、视频抠图、透明视频、BiRefNet、JPG/PNG/BMP/WebP 图片，或 MP4/MOV/WebM 视频的场景；无需 Python、PyTorch 或 CUDA。

aiskillstore/marketplace