深度学习 Skill技能列表
HuggingFace加速器Skill huggingface-accelerate
HuggingFace Accelerate 是一个用于简化分布式训练的Python库,只需添加四行代码即可为PyTorch脚本提供分布式支持,支持DeepSpeed、FSDP等多种后端,自动处理设备放置和混合精度训练,是HuggingFace生态系统中的标准工具,适用于快速原型开发和统一API。关键词:分布式训练、PyTorch、HuggingFace、加速器、DeepSpeed、FSDP、混合精度、AI训练、深度学习。
PyTorch几何库Skill torch-geometric
这个技能是用于开发和训练图神经网络(GNN)的Python库,基于PyTorch。它支持节点分类、图分类、链接预测、异构图处理、分子属性预测等应用,适用于社交网络分析、引用网络、3D几何数据等场景,是几何深度学习的强大工具。关键词:图神经网络、深度学习、PyTorch、GCN、GAT、GraphSAGE、节点分类、图分类、链接预测、分子属性预测、异构图。
AlphaFold数据库Skill alphafold-database
AlphaFold 数据库技能用于访问 AI 预测的蛋白质结构,支持通过 UniProt ID 检索结构、下载 PDB/mmCIF 文件、分析置信度指标(如 pLDDT 和 PAE),适用于药物发现、结构生物学研究和蛋白质工程。关键词:AlphaFold, 蛋白质结构预测, AI, 生物信息学, 药物研发
PyTorchLightningSkill pytorch-lightning
PyTorch Lightning 是一个深度学习框架,用于将 PyTorch 代码组织成模块化结构,自动化训练流程,支持多 GPU/TPU 训练、数据管理、回调机制、日志记录和分布式训练策略,以实现可扩展的神经网络训练。关键词:深度学习,PyTorch,训练自动化,多设备训练,分布式训练,神经网络,AI框架。