工程开发
at-dispatch-v2
将 PyTorch ATen C++ 代码从旧版 AT_DISPATCH 宏转换为现代 AT_DISPATCH_V2 API 的自动化工具。
浏览: 18★ 243
在找 pytorch 相关工作流吗?这些 agent skills 可协助开发者自动化设置、编程与交付。
探索可复用的代理技能,查看实现细节,快速找到适合你工作流程的技能。
共找到 6 个技能
将 PyTorch ATen C++ 代码从旧版 AT_DISPATCH 宏转换为现代 AT_DISPATCH_V2 API 的自动化工具。
更新 PyTorch AT_DISPATCH_V2 宏,为运算符和内核启用 uint16、uint32 与 uint64 支持。
使用 FlashInfer API 日志功能排查与解决 CUDA 运行时崩溃的教学指南。
使用 Stable Baselines3 进行生产级强化学习。通过类 scikit-learn API 训练智能体、设计自定义环境、实现训练回调函数并优化工作流程。
PyTorch Lightning 深度学习框架技能:自动化模型训练、多 GPU 编排、数据管道以及 DDP、FSDP 和 DeepSpeed 等分布式训练策略。
根据 PyTorch 惯例与 Sphinx 格式,自动生成或更新函数与方法的 docstring 文件说明。