at-dispatch-v2
Automated conversion tool for migrating PyTorch ATen C++ code from legacy AT_DISPATCH macros to the modern AT_DISPATCH_V2 API.
Looking for pytorch workflows? These agent skills help developers automate setup, coding, and delivery tasks.
Discover reusable agent skills, browse implementation details, and find the right skill for your workflow.
6 skills found
Automated conversion tool for migrating PyTorch ATen C++ code from legacy AT_DISPATCH macros to the modern AT_DISPATCH_V2 API.
Update PyTorch AT_DISPATCH_V2 macros to enable uint16, uint32, and uint64 support in operators and kernels.
Tutorial for identifying and resolving CUDA runtime crashes using FlashInfer's API logging framework.
Production-ready reinforcement learning using Stable Baselines3. Train agents, design custom environments, implement training callbacks, and optimize workflows with a scikit-learn-style API.
PyTorch Lightning skill for scalable deep learning: automates model training, multi-GPU orchestration, data pipelines, and distributed training strategies like DDP, FSDP, and DeepSpeed.
Generate and update PyTorch-compliant function and method docstrings using reStructuredText/Sphinx conventions.