torchaudio

Torchaudio 是一个用于 PyTorch 的音频和信号处理库。它提供 I/O、信号和数据处理函数、数据集、模型实现和应用程序组件。

torchtune

一个专为微调大型语言模型 (LLM) 而设计的 PyTorch 原生库。 torchtune 支持完整的微调工作流程，并与流行的生产推理系统兼容。

torchvision

此库是 PyTorch 项目的一部分。 PyTorch 是一个开源机器学习框架。 torchvision 包包含常用的数据集、模型架构和用于计算机视觉的常见图像转换。

torcharrow

TorchArrow 是一个类似 torch.Tensor 的 Python DataFrame 库，用于深度学习中的数据预处理。 它支持多种执行运行时和 Arrow 作为通用格式。

TorchData

一个 Beta 库，包含用于轻松构建灵活且高性能数据管道的常用模块化数据加载原语。 并且，还有一些功能仍处于原型阶段。

TorchRec

TorchRec 是一个 PyTorch 领域库，旨在提供大规模推荐系统 (RecSys) 所需的常用稀疏性和并行性原语。 它允许作者训练在多个 GPU 之间分片的具有大型嵌入表的模型。

TorchServe

TorchServe 是一个高性能、灵活且易于使用的工具，用于在生产环境中服务 PyTorch 模型。

XLA 设备上的 PyTorch

PyTorch 通过 torch_xla 软件包在 XLA 设备（如 TPU）上运行。 本文档介绍了如何在这些设备上运行模型。

torchrl

torchrl 是一个开源的、以 Python 为先的 PyTorch 强化学习库，专注于高模块化和良好的运行时性能，为成本函数、回报和数据处理提供低级和高级 RL 抽象和可重用的函数。

tensordict

tensordict 是一个类似字典的类，它简化了张量批次的操作，通过抽象定制操作并通过叶子上的自动操作分派来减少错误，从而增强了代码的可读性、紧凑性和模块化。

torchtext

此库是 PyTorch 项目的一部分。 PyTorch 是一个开源机器学习框架。 torchtext 包包含用于自然语言的数据处理实用程序和常用数据集。

ExecuTorch

ExecuTorch 是一个 PyTorch 平台，提供基础设施，可在从 AR/VR 可穿戴设备到标准设备端 iOS 和 Android 移动部署等各种环境中运行 PyTorch 程序。

torchao

torchao 是一个用于自定义数据类型和优化的库。 使用原生 PyTorch 量化和稀疏化权重、梯度、优化器以及用于推理和训练的激活。