搜索结果：pytorch（29 个）
按分类缩小范围
datasets
综合 76
汇聚海量 AI 模型开箱即用数据集的枢纽，并提供快速、易用且高效的数据处理工具。
stable-baselines3
综合 75
基于 PyTorch 实现的强化学习算法库，提供可靠的算法实现。
ludwig
综合 73
用于构建自定义 LLM、神经网络及其他 AI 模型的低代码框架。
peft
综合 71
Hugging Face 推出的参数高效微调（PEFT）库，助你低成本微调大模型。
ml-engineering
综合 71
机器学习工程开源电子书。
speechbrain
综合 70
基于 PyTorch 的开源全能语音处理工具包。
onnx
综合 69
用于机器学习模型互操作性的开放标准。
pytorch-grad-cam
综合 69
计算机视觉 AI 可解释性工具，支持 CNN、Vision Transformer 等模型及分类、检测等任务。
pytorch_geometric
综合 67
基于 PyTorch 的图神经网络（GNN）开源库。
autogluon
综合 67
仅需三行代码即可实现的快速且高精度的 AutoML 机器学习库。
vision
综合 65
PyTorch 专用于计算机视觉的数据集、数据变换工具和预训练模型库。
flair
综合 64
极简易用的前沿自然语言处理（NLP）开发框架。
yolov3
综合 64
YOLOv3 系列的 PyTorch 实现，支持实时目标检测的训练、验证、推理及多格式导出
nerfstudio
综合 63
一个对协作友好的 NeRF（神经辐射场）开发平台。
deeplearning4j
综合 63
基于 JVM 的深度学习训练与部署工具套件，支持导入 Keras、TensorFlow、ONNX/PyTorch 等模型。
使用 PyTorch 从零开始、循序渐进实现类似 ChatGPT 的大语言模型教程。
ivy
综合 63
在不同框架间转换机器学习代码。
tianshou
综合 61
基于 PyTorch 构建的优雅、灵活的深度强化学习库。
open_clip
综合 61
CLIP 的开源实现。
nano-vllm
综合 61
Nano vLLM