Is it compliant with official PyTorch standards?

Yes, it follows the conventions established in core PyTorch files like torch/_tensor_docs.py and torch/nn/functional.py.

Can it link to other PyTorch modules?

Yes, the skill includes formatting rules for cross-referencing classes, functions, and methods using Sphinx roles like :class:, :func:, and :meth:.

Does this skill support LaTeX math?

Yes, it specifically uses raw strings and Sphinx math directives to ensure complex deep learning formulas are rendered correctly without backslash escape issues.

Does it generate usage examples?

Absolutely. One of the core requirements of this skill is to include an 'Examples::' section with properly formatted Python prompts and outputs.

PyTorch Docstring Generator

Name: PyTorch Docstring Generator
Author: biobenkj

bybiobenkj

0•

Learning & Documentation

Writes and updates professional-grade docstrings for PyTorch functions and methods following official Sphinx and reStructuredText conventions.

This skill provides Claude with specific expertise in drafting high-quality documentation for PyTorch-based projects. It enforces the strict formatting standards found in official PyTorch source code, including the use of raw strings for LaTeX math compatibility, standardized function signatures, and structured sections for arguments, return types, and usage examples. It is particularly useful for deep learning researchers and engineers who need to maintain clear, documentation-site-ready code that includes complex mathematical formulas and cross-references to other PyTorch modules.

Key Features

010 GitHub stars

02Enforces Sphinx and reStructuredText (reST) formatting standards

03Standardizes Args, Keyword Args, and Returns sections with type annotations

04Ensures docstrings include functional code examples with proper prompt formatting

05Handles complex LaTeX mathematical formulas using math directives

06Automates cross-referencing to PyTorch classes, functions, and methods

Use Cases

01Adding mathematical definitions and tensor shape documentation to deep learning code

02Standardizing documentation across a large-scale PyTorch research library

03Documenting custom neural network layers and functional operations

What are Skills?·How to Install

Install with 🐟 Skill.Fish

npx skillfish add biobenkj/torch-semimarkov docstring

For use in Claude.ai and ChatGPT

Download Skill