PyTorch agent skills, page 3

Guides reinforcement-learning research with torchforge, Meta's PyTorch-native library that keeps RL algorithms apart from infrastructure, including GRPO math-reasoning runs.

Orchestra-Research/AI-Research-SKILLs

Sets up large-scale LLM training with NVIDIA Megatron-Core, choosing tensor, pipeline, data, context and expert parallelism for a given model size and GPU count.

Orchestra-Research/AI-Research-SKILLs

Step-by-step playbook for migrating a pytorch/pytorch .github/workflows/.yml from EC2 to OSDC (ARC) runners — covers both dial-up and 100% opt-in patterns, with the inputs that must be plumbed…

pytorch/test-infra

QAI ModelBuilder. An agent skill from qualcomm/qai-appbuilder.

qualcomm/qai-appbuilder

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/Auto-claude-code-research-in-sleep

Guide for using MPK test mode to unit-test individual layers or multi-layer pipelines through the full compilation pipeline.

mirage-project/mirage

将原生 PyTorch 自定义算子库、Torch extension、生态库（TorchCodec/FlashInfer/DeepEP 等）以及 Kernel DSL 生态（Triton/TileLang/TVM FFI 等）以最小修改方式接入 PaddlePaddle。遇到以下场景务必使用：迁移外部算子库到 Paddle；分析 PFCCLab fork 与上游的兼容差异；处理…

PaddlePaddle/Paddle

Organizes PyTorch training code into LightningModules, DataModules and Trainers, with multi-GPU strategies, callbacks and logging configured.

davila7/claude-code-templates

Triages GitHub issues by routing to oncall teams, applying labels, and closing questions.

pytorch/pytorch

Checks training code, configs and math against documented framework behavior before an expensive run, citing a knowledge base or official docs for every claim.

Leeroo-AI/superml

Analyze single-file PyTorch/Kineto Chrome trace .json(.gz) files using the VeloQ CLI.

lucifer1004/VeloQ

Run, resume, monitor, diagnose, and report Quark Quant-Perf workflows for PyTorch and HuggingFace transformers models.

amd/Quark

Explains machine learning predictions with SHAP: picking the right explainer, computing Shapley values and drawing waterfall, beeswarm, bar and force plots.

davila7/claude-code-templates

PyTorch training reference: architecture choice by data type, scaling rules, a training loop, optimizer and learning-rate choices, and fixes for loss spikes or OOM.

Orchestra-Research/AI-Research-SKILLs

Detects CPU, GPU, memory and disk resources before heavy scientific tasks and writes a JSON file with advice on parallelism, out-of-core work and GPU use.

davila7/claude-code-templates

Export a PyTorch model to .pte format for ExecuTorch. Use when converting models, lowering to edge, or generating .pte files.

pytorch/executorch

Covers logging and viewing training metrics, histograms, model graphs, embeddings and profiles with TensorBoard in PyTorch and TensorFlow projects.

Orchestra-Research/AI-Research-SKILLs

Guide to using Mamba selective state-space models for linear-time sequence modeling, from the Mamba block and pretrained checkpoints to Mamba-2 and speed comparisons.

Orchestra-Research/AI-Research-SKILLs

Walks through nanoGPT, Karpathy's compact GPT implementation: training on Shakespeare, reproducing GPT-2, fine-tuning GPT-2 checkpoints and training on your own text.

Orchestra-Research/AI-Research-SKILLs

Explains RWKV, a hybrid that trains in parallel like a GPT and runs inference like an RNN with constant memory per token, plus usage, fine-tuning and troubleshooting.

Orchestra-Research/AI-Research-SKILLs

Load this FIRST whenever working with PyTorch CI data (any pytorch/ org repo), the torchci/HUD codebase, or the PyTorch HUD ClickHouse database.

pytorch/test-infra

Publish a Python package to PyPI using uv with credentials loaded from a local .secrets file.

ML4ITS/TimeVQVAE

Guide to renting GPUs on Lambda Labs for ML training and inference: on-demand instances, 1-Click Clusters, SSH access, persistent filesystems and alternatives.

Orchestra-Research/AI-Research-SKILLs

Graph-based drug discovery toolkit. An agent skill from davila7/claude-code-templates.

davila7/claude-code-templates

Cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry.

davila7/claude-code-templates

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

Orchestra-Research/AI-Research-SKILLs

Expert in PerforatedAI library for adding artificial dendrites to PyTorch neural networks.

PerforatedAI/PerforatedAI

Graph Neural Networks (PyG). An agent skill from davila7/claude-code-templates.

davila7/claude-code-templates

PyTorch deep learning patterns and best practices for building robust, efficient, and reproducible training pipelines, model architectures, and data loading.

affaan-m/ECC

Query CZ CELLxGENE Census (61M+ cells). An agent skill from davila7/claude-code-templates.

davila7/claude-code-templates

Import PyTorch, ONNX, or Keras 3 / TensorFlow 2.16+ deep learning models into MATLAB as dlnetwork objects.

matlab/matlab-agentic-toolkit

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

Author a new ShapeShifter graph-transformation pass for AMD Quark (ONNX or PyTorch) so it conforms to the pass framework's conventions and auto-registers.

amd/Quark

End-to-end AscendC custom operator development for Ascend NPU in an ascend-kernel (csrc/ops + build.sh + torchnpu PyTorch custom op) project.

ascend-ai-coding/awesome-ascend-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

Build production computer vision pipelines for object detection, tracking, and video analysis.

curiositech/some_claude_skills

Pick the right ComfyUI startup flags for VRAM, attention, caching, and speed.

artokun/comfyui-mcp

how to use the edit command properly

omegaml/omegaml

Disable or re-enable Cloudflare R2 (download-r2.pytorch.org) usage in managev2.py during R2 outages.

pytorch/test-infra

Profile ExecuTorch model execution. Use when measuring performance, analyzing operator timing, or debugging slow models.

pytorch/executorch

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

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

uw-syfi/vibesys

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

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

davila7/claude-code-templates