工程開發
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 文件說明。