Torch Python Packages

torch-pitch-shift

Pitch-shift audio clips quickly with PyTorch (CUDA supported)! Additional utilities for searching efficient transformations are included.

1.2M 139 11

torchinfo

View model summaries in PyTorch!

422K 3K 140

gluonts

Probabilistic time series modeling in Python

290K 5K 841

torchft-nightly

Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)

133K 543 80

kubeflow

Universal Python SDK to run AI workloads on Kubernetes

119K 155 286

skrl

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

101K 1K 159

zuko

Normalizing flows in PyTorch

98K 466 40

torchrl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

91K 4K 502

torchrl-nightly

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

60K 4K 502

ibverbs

High-performance RDMA for Python and PyTorch (ibverbs, efa, NVMe-oF)

24K 13 2

metatrain

Train, fine-tune, and manipulate machine learning models for atomistic systems

22K 80 41

efa

High-performance RDMA for Python and PyTorch (ibverbs, efa, NVMe-oF)

19K 13 2

pydantic-tensor

Pydantic support for parsing, validation, and serialization of tensors

17K 8 1

alias-free-torch

Simple torch.nn.module implementation of Alias-Free-GAN style filter and resample

17K 103 11

metal-linalg

QR, symmetric eigendecomposition and SVD for batches of matrices on Apple Silicon GPUs, for MLX and PyTorch. 2x to 10x gain in performance. Metal kernels per regime, routed to the GPU or the CPU by policies measured per Mac. C++, Python (pip: metal-linalg, metal-linalg-torch), Swift and C.

17K 14 0