torchaudio

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

torchtune

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 通过 torch_xla 包在 TPUs 等 XLA 设备上运行。本文档描述了如何在这些设备上运行您的模型。

torchrl

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

tensordict

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

torchtext

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

ExecuTorch

ExecuTorch 是一个 PyTorch 平台，提供基础设施以在从 AR/VR 可穿戴设备到标准的设备上 iOS 和 Android 移动部署的任何地方运行 PyTorch 程序。

torchao

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