下载

Skill UI

浏览并发现 11126+ 精选技能

全部编程开发人工智能设计创意产品商业数据科学市场营销职场通用效率工具硬件工程语言学习

搜索 PyTorch ，共找到 45 条记录

默认排序最新上传最多下载

PyTorch Lightning深度学习框架

pytorch-lightning

K-Dense-AI/scientific-agent-skills

PyTorch Lightning是一个全面的深度学习框架，用于组织和自动化PyTorch模型的训练流程。它通过将模型组织为LightningModules，简化了繁琐的编写代码，并在Trainer中自动化训练循环。该框架支持多GPU/TPU分布式训练（DDP, FSDP），并集成了数据加载、回调函数和实验跟踪（W&B, MLflow），适用于构建专业、可扩展的深度学习项目。

稳定强化学习算法库

stable-baselines3

K-Dense-AI/scientific-agent-skills

Stable Baselines3是一个基于PyTorch的专业级强化学习框架。它提供了PPO、SAC、DQN等一系列稳定且可靠的RL算法实现。用户可以使用它进行标准RL实验、快速原型开发，并支持构建和训练复杂的单智能体系统，包括自定义和向量化环境。

使用PyTorch构建图神经网络

torch-geometric

K-Dense-AI/scientific-agent-skills

PyTorch Geometric (PyG) 是基于PyTorch的专业图神经网络（GNN）标准库。它提供了一系列用于处理各类图数据（包括异构图）的专业数据结构，并内置了超过60种GNN层实现（如GCN、GAT）。该库支持完整的图数据集加载和数据预处理，是进行节点分类、图分类和链路预测等深度学习图任务的核心工具。

药物发现深度学习工具箱

K-Dense-AI/scientific-agent-skills

TorchDrug是一个基于PyTorch的综合性机器学习工具箱，专注于分子和蛋白质科学。它利用图神经网络（GNNs）处理复杂的生物和化学任务，包括分子属性预测、蛋白质结构建模、药物靶点结合预测、知识图谱推理和新分子生成，旨在加速药物发现研究。

CoreWeave分布式GPU训练工作流

coreweave-core-workflow-b

jeremylongshore/claude-code-plugins-plus-skills

本指南详细介绍了如何在CoreWeave平台上运行大规模的分布式GPU训练任务。它涵盖了单节点多GPU配置和多节点训练，支持PyTorch DDP，是进行大型语言模型（LLM）微调或需要高性能计算集群的深度学习模型训练的推荐工作流。

实时深度估计隐私处理

depth-estimation

SharpAI/DeepCamera

本技能利用Depth Anything v2模型实现实时单目深度估计，能够将实时摄像头画面转化为彩色深度图。它提供深度图层叠加功能，并特别支持“隐私模式”，通过深度数据处理来完全匿名化场景中的人物和细节，但能保留完整的空间布局和活动轨迹，适用于安全监控和隐私保护领域。

KubeSphere Volcano作业编排

kubesphere-volcano

kubesphere/kubesphere

这是一个用于KubeSphere环境下的Volcano作业管理工具。它支持用户管理整个作业的生命周期，包括创建、列表、更新和删除Job。特别适用于深度学习（PyTorch/TensorFlow）、MPI以及通用批处理任务的调度和故障排除，提供详细的API操作指南。

深度学习面试知识速查手册

interview-cheatsheet

wanshuiyin/Auto-claude-code-research-in-sleep

该技能可根据指定的机器学习或大型语言模型（LLM）主题，生成一份长篇、结构极其严谨的中文教程速查手册。教程不仅包含核心公式推导和从零开始的PyTorch代码实现，还设有详细的对比表格和25个分级（L1-L3）高频面试题，是进行深度学术学习和技术面试准备的理想工具。

PyTorch深度学习模式指南

pytorch-patterns

本技能旨在指导用户掌握 PyTorch 框架中的常见模式和设计最佳实践。它适用于开发复杂的深度学习模型，有助于提升神经网络的结构化程度、代码的复用性以及整体的工程效率。

上一页 1 2 34

语言