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 模型的高性能、灵活且易于使用的工具。

PyTorch 在 XLA 设备上

PyTorch 使用 torch_xla 包在 XLA 设备（如 TPU）上运行。本文档介绍了如何在这些设备上运行您的模型。

torchrl

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

tensordict

tensordict 是一种类似字典的类，它简化了对张量批次的运算，通过抽象定制运算并通过自动运算分派减少错误，从而提高代码的可读性、紧凑性和模块化。

torchtext

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