PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Torch Python Packages

Python packages with the GitHub topic torch. Sorted by relevance, with stars and monthly downloads.
awslabs
gluonts

Probabilistic time series modeling in Python

3.5M 5K 818
KentoNishi
torch-pitch-shift

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

1.7M 139 12
pytorch
torchrl

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

1.4M 3K 450
tyleryep
torchinfo

View model summaries in PyTorch!

866K 3K 136
junjun3518
alias-free-torch

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

231K 100 10
pytorch
torchft-nightly

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

219K 500 62
kubeflow
kubeflow

Universal Python SDK to run AI workloads on Kubernetes

210K 112 174
opentensor
bittensor

Internet-scale Neural Networks

163K 1K 468
Toni-SM
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

74K 1K 141
zh217
torch-dct

DCT (discrete cosine transform) functions for pytorch

64K 642 75
bshillingford
torchfile

Deserialize Lua torch-serialized objects from Python

42K 217 24
pytorch
torchrl-nightly

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

40K 3K 450
tyleryep
torch-summary

View model summaries in PyTorch!

37K 3K 136
szymonmaszke
torchdatasets-nightly

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

30K 331 20
mberr
torch-max-mem

Decorators for maximizing memory utilization with PyTorch & CUDA

28K 17 0
pykeen
pykeen

🤖 A Python library for learning and evaluating knowledge graph embeddings

27K 2K 218
probabilists
zuko

Normalizing flows in PyTorch

25K 457 36
Artificial-Sweetener
torchlanc

A High-Quality Lanczos Scaler for Pytorch

22K 2 0
dominik-schwabe
pydantic-tensor

Pydantic support for parsing, validation, and serialization of tensors

22K 8 1
DPIRD-DMA
omnicloudmask

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

13K 169 20
uofgravity
glasflow

Normalising flows implemented using nflows

12K 26 3
metatensor
metatrain

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

10K 68 26
OPHoperHPO
carvekit

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

9K 2K 283
2noise
chattts

A generative speech model for daily dialogue.

8K 39K 4K
    • Data from PyPI, GitHub, ClickHouse, and BigQuery