Torch Python Packages

torch-pitch-shift

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

gluonts

Probabilistic time series modeling in Python

torchrl

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

torchinfo

View model summaries in PyTorch!

torchft-nightly

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

kubeflow

Universal Python SDK to run AI workloads on Kubernetes

bittensor

Internet-scale Neural Networks

alias-free-torch

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

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

torch-dct

DCT (discrete cosine transform) functions for pytorch

torch-summary

View model summaries in PyTorch!

torchrl-nightly

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

torch-max-mem

Decorators for maximizing memory utilization with PyTorch & CUDA

pykeen

🤖 A Python library for learning and evaluating knowledge graph embeddings

torchdatasets-nightly

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

zuko

Normalizing flows in PyTorch

torchfile

Deserialize Lua torch-serialized objects from Python

pydantic-tensor

Pydantic support for parsing, validation, and serialization of tensors

torchlanc

A High-Quality Lanczos Scaler for Pytorch

metatrain

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

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. ✂️

glasflow

Normalising flows implemented using nflows

chattts

A generative speech model for daily dialogue.