pytorch
6 skills with substantive source documentation and an official publisher or observed installation demand. Read each skill’s original documentation through its card.
pytorchadd-uint-support
Add unsigned integer (uint) type support to PyTorch operators by updating ATDISPATCH macros. Use when adding support for uint16, uint32, uint64 types to operators, kernels, or when user mentions enabling unsigned types, barebones unsigned types, or uint support.
- 導入数
- 2
- GitHub Stars
- 10.4万
- 更新日
- 10月4日
pytorchat-dispatch-v2
Convert PyTorch ATDISPATCH macros to ATDISPATCHV2 format in ATen C++ code. Use when porting ATDISPATCHALLTYPESAND, ATDISPATCHFLOATINGTYPES, or other dispatch macros to the new v2 API. For ATen kernel files, CUDA kernels, and native operator implementations.
- 導入数
- 1
- GitHub Stars
- 10.3万
- 更新日
- 9月24日
pytorchdocstring
Write docstrings for PyTorch functions and methods following PyTorch conventions. Use when writing or updating docstrings in PyTorch code.
- 導入数
- 1
- GitHub Stars
- 10.3万
- 更新日
- 9月24日
pytorchmetal-kernel
Write Metal/MPS kernels for PyTorch operators. Use when adding MPS device support to operators, implementing Metal shaders, or porting CUDA kernels to Apple Silicon. Covers nativefunctions.yaml dispatch, host-side operators, and Metal kernel implementation.
- 導入数
- 1
- GitHub Stars
- 10.3万
- 更新日
- 9月24日
pytorchpyrefly-type-coverage
Migrate a file to use stricter Pyrefly type checking with annotations required for all functions, classes, and attributes.
- 導入数
- 1
- GitHub Stars
- 10.3万
- 更新日
- 9月24日
pytorchtriaging-issues
Triages GitHub issues by routing to oncall teams, applying labels, and closing questions. Use when processing new PyTorch issues or when asked to triage an issue.
- 導入数
- 1
- GitHub Stars
- 10.3万
- 更新日
- 9月24日