登录
下载

Skill UI

浏览并发现 10321+ 精选技能
全部 编程开发 人工智能 设计创意 产品商业 数据科学 市场营销 职场通用 效率工具 硬件工程 语言学习
搜索 PyTorch ，共找到 44 条记录
默认排序 最新上传 最多下载

PyTorch FSDP2 训练流程

pytorch-fsdp2
Orchestra-Research/AI-Research-SKILLs
439
指导在训练脚本中正确引入 PyTorch FSDP2 fully_shard，使底层模块逐层分片、配置混合精度与 CPU 卸载、构建依赖 DTensor 的优化器，并依靠分布式检查点处理超过单卡显存或需要 DeviceMesh 分片的模型。
查看详情

PyTorch Lightning 训练框架

pytorch-lightning
Orchestra-Research/AI-Research-SKILLs
235
PyTorch Lightning 提供高层训练框架，内置分布式训练（DDP/FSDP/DeepSpeed）、回调、检查点、日志与设备管理，帮助你从笔记本到多节点集群轻松运行高质量模型训练。
查看详情

PyTorch模型训练助手

pytorch-model-trainer
jeremylongshore/claude-code-plugins-plus-skills
381
自动化 PyTorch 训练流程，提供数据准备、模型配置、超参调优与实验跟踪等建议，遵循行业最佳实践，适合需要快速搭建训练流水的使用场景。
查看详情

PyTorch模型因果干预

pyvene-interventions
Orchestra-Research/AI-Research-SKILLs
110
pyvene是一个用于在PyTorch模型上执行因果干预的声明式框架。它支持激活打补丁、因果追踪（ROME风格）和交替干预训练等高级实验。适用于需要测试模型因果假设、深入理解模型组件工作机制或确保可复现性的AI研究场景。
查看详情

Ray数据处理平台

ray-data
Orchestra-Research/AI-Research-SKILLs
76
Ray Data 提供可扩展的分布式数据处理，支持 CPU/GPU 流式执行，兼容 Parquet、CSV、JSON、图像等多模态，集成 PyTorch、TensorFlow、Ray Train，适合批量推理、ETL 与大规模预处理。
查看详情

Ray Train 分布式协调

ray-train
Orchestra-Research/AI-Research-SKILLs
64
Ray Train 统一协调 PyTorch、TensorFlow 与 HuggingFace 的分布式训练，从单机扩展到多节点集群，自动处理显卡分配、容错、断点恢复与超参搜索，帮助团队在无需大量改动的前提下完成大规模模型训练。
查看详情

计算机视觉生产级工程技能

senior-computer-vision
alirezarezvani/claude-skills
257
涵盖了从基础模型到生产部署的完整计算机视觉AI全流程技能。掌握目标检测（YOLO、Faster R-CNN）、实例/语义分割（Mask R-CNN、SAM）等核心技术，并熟练运用PyTorch进行模型训练，最终通过ONNX和TensorRT等工具优化模型，实现边缘或云端设备的稳定部署。
查看详情

TensorBoard 可视化工具包

tensorboard
Orchestra-Research/AI-Research-SKILLs
130
TensorBoard 提供统一可视化界面，跟踪训练指标、调试模型、对比实验、观察网络结构并分析性能瓶颈，适用于 PyTorch 与 TensorFlow 训练流程。
查看详情

PyTorch原生智能体RL训练框架

torchforge-rl-training
Orchestra-Research/AI-Research-SKILLs
135
torchforge是Meta推出的用于智能体强化学习（RL）的PyTorch原生框架。它将核心的RL算法与复杂的分布式基础设施进行了彻底分离。用户可以专注于算法的快速实验和开发，无需担心底层分布式训练、权重同步等复杂问题，支持从单卡到大规模集群的扩展训练。
查看详情

训练测试集切分器

train-test-splitter
jeremylongshore/claude-code-plugins-plus-skills
383
该技能自动响应涉及训练/测试集划分的请求，提供行业级指导、实用代码与配置建议，并帮助验证 PyTorch、TensorFlow 或 sklearn 流程中的数据准备与模型训练环节。
查看详情

Vast.ai GPU实例生命周期管理

vastai-hello-world
jeremylongshore/claude-code-plugins-plus-skills
237
本指南全面演示了在Vast.ai上管理GPU计算资源的完整生命周期。内容涵盖使用CLI和REST API搜索可用硬件、配置GPU实例（例如使用PyTorch镜像）、通过SSH连接、运行深度学习工作负载，以及最后销毁实例以停止计费。适用于测试云集成或管理机器学习计算流水线。
查看详情

AI深度学习GPU性能优化

vastai-performance-tuning
jeremylongshore/claude-code-plugins-plus-skills
77
这是一份关于深度学习工作流和云GPU平台性能优化的综合指南。内容涵盖了如何通过性能/金钱比选择最佳实例、减少启动延迟、精细调优数据管道（如DataLoader和pin_memory），以及使用PyTorch和Bash脚本实现多GPU分布式训练，以最大化计算资源的利用率和训练效率。
查看详情
上一页1234下一页
语言
简体中文
English