pytorch

具备强大 GPU 加速的 Python 张量与动态神经网络库

围绕 PyTorch 的视觉、文本、强化学习等示例集合。

针对计算机视觉的数据集、转换与模型

PyTorch 教程

一个 PyTorch 原生平台，用于训练生成式 AI 模型。

在移动、嵌入式与边缘设备上执行的 PyTorch AI

高阶函数库，协助在 PyTorch 中灵活且透明地训练与评估神经网络

一个模块化、原始型优先、Python 首先的 PyTorch Reinforcement Learning 函数库。

使用 TensorRT 为 NVIDIA GPU 编译 PyTorch/TorchScript/FX 的编译器

PyTorch 原生量化，用于训练和推理

基于 PyTorch 的音讯信号处理数据操作与转换

在 XLA 设备（如 Google TPU）上启用 PyTorch

FB (Facebook) + GEMM (通用矩阵-矩阵乘法) - https://code.fb.com/ml-applications/fbgemm/

具有多机训练各种原语的集合通讯库。

提交至 https://pytorch.org/hub/

PyTorch 中的 C++ 扩展功能

CPU 信息库 (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)

TorchBench 是一个用于评估 PyTorch 性能的开源基准测试合集。

TensorDict 是一个专为 pytorch 设计的 tensor 容器。

一个 CPU+GPU 性能分析库，提供时间轴追踪与硬件性能计数器的访问。

一个内嵌于 Python 的 DSL，能以极少的样板代码轻松编写快速、可扩展的 ML 核心。