torchaudio

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

torchtune

torchtune 是一个为大型语言模型微调而设计的 PyTorch 原生库（LLMs）。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 包包含自然语言处理的数据处理工具和常用数据集。

ExecuTorch

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

torchao

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