torchaudio

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

torchtune

一个 PyTorch 原生库，专为微调大型语言模型 (LLM) 而设计。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 在 XLA 设备（如 TPU）上运行，并使用 torch_xla 包。本文档介绍如何在这些设备上运行模型。

torchrl

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

tensordict

tensordict 是一个类似于字典的类，可简化对一批张量的操作，通过抽象定制操作和通过在叶子节点上自动调度操作来减少错误，从而增强代码的可读性、紧凑性和模块化。

torchtext

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