Torch Python Packages

gluonts

Probabilistic time series modeling in Python

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.

torchinfo

View model summaries in PyTorch!

alias-free-torch

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

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

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

torchfile

Deserialize Lua torch-serialized objects from Python

torchrl-nightly

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

torch-summary

View model summaries in PyTorch!

torchdatasets-nightly

PyTorch based library focused on data processing and input pipelines in general.

torch-max-mem

Decorators for maximizing memory utilization with PyTorch & CUDA

pykeen

🤖 A Python library for learning and evaluating knowledge graph embeddings

zuko

Normalizing flows in PyTorch

torchlanc

A High-Quality Lanczos Scaler for Pytorch

pydantic-tensor

Pydantic support for parsing, validation, and serialization of tensors

omnicloudmask

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

glasflow

Normalising flows implemented using nflows

metatrain

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

carvekit

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

chattts

A generative speech model for daily dialogue.