Torch Python Packages

torch-pitch-shift

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

torchrl

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

gluonts

Probabilistic time series modeling in Python

torchinfo

View model summaries in PyTorch!

bittensor

Internet-scale Neural Networks

kubeflow

Universal Python SDK to run AI workloads on Kubernetes

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

torchft-nightly

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

alias-free-torch

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

torch-summary

View model summaries in PyTorch!

zuko

Normalizing flows in PyTorch

torch-dct

DCT (discrete cosine transform) functions for pytorch

torchrl-nightly

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

metatrain

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

pykeen

🤖 A Python library for learning and evaluating knowledge graph embeddings

torch-max-mem

Decorators for maximizing memory utilization with PyTorch & CUDA

pydantic-tensor

Pydantic support for parsing, validation, and serialization of tensors

torchdatasets-nightly

PyTorch dataset extended with map, cache etc. (tensorflow.data like)

torchlanc

A High-Quality Lanczos Scaler for Pytorch

glasflow

Normalising flows implemented using nflows

omnicloudmask

OmniCloudMask is a Python library for fast, accurate cloud and cloud shadow segmentation in satellite imagery.

carvekit

✂️ Automated high-quality background removal framework for an image using neural networks. ✂️

chattts

A generative speech model for daily dialogue.

torchfx

A GPU accelerated and torch based audio DSP library