An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.

pypi.org "torch" keyword

View the packages on the pypi.org package registry that are tagged with the "torch" keyword.

qytorch 0.0.3
A library to make quaternion neural networks.
3 versions - Latest release: over 1 year ago - 14 downloads last month - 0 stars on GitHub - 1 maintainer
omnicloudmask 1.6.0
Python library for cloud and cloud shadow segmentation in high to moderate resolution satellite i...
17 versions - Latest release: about 2 months ago - 2.14 thousand downloads last month - 100 stars on GitHub - 1 maintainer
eurocropsml 0.4.1
EuroCropsML is a ready-to-use benchmark dataset for few-shot crop type classification using Senti...
6 versions - Latest release: 6 months ago - 56 downloads last month - 20 stars on GitHub - 1 maintainer
Top 1.3% on pypi.org
gluonts 0.16.2
Probabilistic time series modeling in Python.
111 versions - Latest release: 4 months ago - 18 dependent packages - 77 dependent repositories - 1.99 million downloads last month - 4,258 stars on GitHub - 3 maintainers
sphericart-torch 1.0.5
TorchScript bindings to sphericart
11 versions - Latest release: about 6 hours ago - 1 dependent package - 9.35 thousand downloads last month - 88 stars on GitHub - 3 maintainers
torchconvview 0.2.0
A library for PyTorch convolution layer visualizations.
2 versions - Latest release: about 2 years ago - 15 downloads last month - 2 stars on GitHub - 1 maintainer
Top 6.6% on pypi.org
megatron-core 0.14.0
Megatron Core - a library for efficient and scalable training of transformer based models
88 versions - Latest release: 26 days ago - 3 dependent packages - 1 dependent repositories - 81.5 thousand downloads last month - 13,940 stars on GitHub - 3 maintainers
megatron-fsdp 0.1.0
**Megatron-FSDP** is an NVIDIA-developed PyTorch extension that provides a high-performance imple...
30 versions - Latest release: 26 days ago - 23.9 thousand downloads last month - 13,940 stars on GitHub - 1 maintainer
terge 0.1.1
An easy-to-use Python library for merging PyTorch models.
2 versions - Latest release: over 1 year ago - 9 downloads last month - 10 stars on GitHub - 2 maintainers
abracudabra 0.1.3
Convert Tensors, Arrays and DataFrames Between CPU and CUDA
4 versions - Latest release: 8 months ago - 34 downloads last month - 1 stars on gitlab.cern.ch - 1 maintainer
Top 1.9% on pypi.org
onnx2torch 1.5.15
ONNX to PyTorch converter
22 versions - Latest release: about 1 year ago - 17 dependent packages - 44 dependent repositories - 81 thousand downloads last month - 579 stars on GitHub - 1 maintainer
Top 2.8% on pypi.org
torchrl 1.8.4
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning
45 versions - Latest release: over 6 years ago - 6 dependent packages - 16 dependent repositories - 148 thousand downloads last month - 1,571 stars on GitHub - 1 maintainer
Top 1.7% on pypi.org
torchinfo 1.8.0 💰
Model summary in PyTorch, based off of the original torchsummary.
32 versions - Latest release: over 2 years ago - 67 dependent packages - 489 dependent repositories - 544 thousand downloads last month - 2,476 stars on GitHub - 1 maintainer
stable-fast 1.0.5 💰
Stable Fast is an ultra lightweight performance optimization framework for Hugging Fase diffuser ...
27 versions - Latest release: over 1 year ago - 2 dependent packages - 179 downloads last month - 1,287 stars on GitHub - 1 maintainer
nemo-evaluator 0.1.23
Common utilities for NVIDIA evaluation frameworks
25 versions - Latest release: about 22 hours ago - 16.2 thousand downloads last month - 34 stars on GitHub - 1 maintainer
nemo-evaluator-launcher 0.1.25
Launcher for the evaluations provided by NeMo Evaluator containers with different runtime backends
33 versions - Latest release: about 22 hours ago - 4.83 thousand downloads last month - 34 stars on GitHub - 1 maintainer
xllm 0.3.24
Simple & Cutting Edge LLM Finetuning
40 versions - Latest release: about 2 years ago - 1 dependent repositories - 70 downloads last month - 259 stars on GitHub - 1 maintainer
Top 1.6% on pypi.org
nemo-toolkit 2.5.2
NeMo - a toolkit for Conversational AI
99 versions - Latest release: 7 days ago - 4 dependent packages - 27 dependent repositories - 727 thousand downloads last month - 11,448 stars on GitHub - 3 maintainers
vak 1.0.4
A neural network framework for researchers studying acoustic communication
47 versions - Latest release: 4 months ago - 1 dependent package - 1 dependent repositories - 231 downloads last month - 67 stars on GitHub - 1 maintainer
deepneighbor 0.3.1
embedding-based item nearest neighborhoods extraction
19 versions - Latest release: about 5 years ago - 1 dependent repositories - 76 downloads last month - 4 stars on GitHub - 1 maintainer
torch-lap-cuda 0.0.3
PyTorch wrapper for HyLAC CUDA library for solving linear assignment problems.
3 versions - Latest release: 3 months ago - 32 downloads last month - 7 stars on GitHub - 1 maintainer
idlmav 1.1.0
Create interactive visualizations of model architectures for display in Jupyter notebooks
2 versions - Latest release: 8 months ago - 17 downloads last month - 2 stars on GitHub - 1 maintainer
auto1111sdk 0.0.95
SDK for Automatic 1111.
14 versions - Latest release: over 1 year ago - 485 downloads last month - 139,162 stars on GitHub - 1 maintainer
torchft-nightly 2025.11.2
Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
261 versions - Latest release: 1 day ago - 9.43 thousand downloads last month - 422 stars on GitHub - 1 maintainer
vigorvision 2.1.3
Advanced AI vision and object detection framework by Vigor Industries Limited.
21 versions - Latest release: 1 day ago - 1.78 thousand downloads last month - 1 maintainer
torch-max-mem 0.1.4 💰
Maximize memory utilization with PyTorch.
10 versions - Latest release: 7 months ago - 1 dependent package - 4 dependent repositories - 29.8 thousand downloads last month - 17 stars on GitHub - 1 maintainer
neuralpde 0.0.1
Neuralpde combines differentiable ode solvers with CNNs to learn PDEs using the method of lines
1 version - Latest release: over 1 year ago - 14 downloads last month - 1 maintainer
raydl 0.0.10
The library of utilities to help you deal with the deep learning in PyTorch.
10 versions - Latest release: over 2 years ago - 44 downloads last month - 1 stars on GitHub - 1 maintainer
opengeokube-mlkit 2023.8a0
MLKit - A quick way to start with machine and deep learning
1 version - Latest release: about 2 years ago - 7 downloads last month - 1 stars on GitHub - 1 maintainer
torchpdlp 0.1.2
Uses a PyTorch-implemented restarted PDHG algorithm with enhancements to solve primal-dual linear...
2 versions - Latest release: 3 months ago - 247 downloads last month - 1 stars on GitHub - 2 maintainers
torch-bspline 0.1.3
A simple library for using b-splines in pytorch.
4 versions - Latest release: 9 months ago - 102 downloads last month - 0 stars on GitHub - 1 maintainer
torchsynth 1.0.2
A modular synthesizer in pytorch, GPU-optional and differentiable
9 versions - Latest release: about 3 years ago - 1 dependent repositories - 111 downloads last month - 359 stars on GitHub - 2 maintainers
cvx2 0.6.0
Tools of CV(Computer Vision)
40 versions - Latest release: 5 months ago - 46 downloads last month
torchrtm 1.2.0
Torch-based Vegetation Radiative Transfer Model library (PROSPECT, SAIL, SMAC)
10 versions - Latest release: 2 days ago - 36 downloads last month - 0 stars on GitHub - 1 maintainer
tdfl 0.1.15
DataFrame supporting PyTorch Tensor columns.
16 versions - Latest release: about 1 year ago - 46 downloads last month - 1 maintainer
Top 9.5% on pypi.org
torchwordemb 0.0.9
Load pretrained word embeddings (word2vec, glove format) into torch.FloatTensor of PyTorch
3 versions - Latest release: over 6 years ago - 1 dependent package - 9 dependent repositories - 65 downloads last month - 88 stars on GitHub - 1 maintainer
dataclass-array 1.5.2
Dataclasses that behave like numpy arrays (with indexing, slicing, vectorization).
12 versions - Latest release: over 1 year ago - 1 dependent repositories - 4.98 thousand downloads last month - 52 stars on GitHub - 1 maintainer
pykaira 0.2.1
A PyTorch-based toolkit for simulating communication systems.
4 versions - Latest release: 5 months ago - 46 downloads last month - 53 stars on GitHub - 1 maintainer
rllib 0.0.1
A lib for RL
1 version - Latest release: over 4 years ago - 1 dependent package - 75 dependent repositories - 677 downloads last month - 0 stars on GitHub - 1 maintainer
Top 3.7% on pypi.org
lit-nlp 1.3.1
🔥LIT: The Learning Interpretability Tool
20 versions - Latest release: 11 months ago - 2 dependent packages - 6 dependent repositories - 12.4 thousand downloads last month - 3,595 stars on GitHub - 6 maintainers
easyrec 0.1.0b0
deep learning frame for recommendation algorithm
3 versions - Latest release: over 4 years ago - 1 dependent repositories - 21 downloads last month - 1 maintainer
tokenlearn 0.2.1
Pre-train Static Embedders
5 versions - Latest release: 5 months ago - 51 downloads last month - 85 stars on GitHub - 1 maintainer
tinder 0.1.9
Pytorch helpers and utils
9 versions - Latest release: about 5 years ago - 1 dependent repositories - 126 downloads last month - 5 stars on GitHub - 1 maintainer
tensorflow-hooks 1.0.0
PyTorch-like hooks for TensorFlow Keras layers
1 version - Latest release: over 1 year ago - 10 downloads last month - 1 stars on GitHub - 1 maintainer
py-img-gen 0.1.0
A library for the book "Image Generation with Python"
2 versions - Latest release: 9 months ago - 506 downloads last month - 2 stars on GitHub - 1 maintainer
celux 0.7.3
Lightspeed video decoding directly into tensors!
54 versions - Latest release: 3 months ago - 300 downloads last month - 15 stars on GitHub - 1 maintainer
mixturelib 0.4.0
mixture lib, python package
6 versions - Latest release: almost 5 years ago - 1 dependent repositories - 17 downloads last month - 2 stars on GitHub - 1 maintainer
Top 4.0% on pypi.org
gempy 2025.2.0 💰
An Open-source, Python-based 3-D structural geological modeling software.
49 versions - Latest release: 4 months ago - 2 dependent packages - 5 dependent repositories - 3.27 thousand downloads last month - 1,154 stars on GitHub - 3 maintainers
ablator 0.0.1b3
Model Ablation Tool-Kit
4 versions - Latest release: almost 2 years ago - 162 downloads last month - 35 stars on GitHub - 1 maintainer
torchdrivesim 0.2.3
TorchDriveSim is a lightweight 2D driving simulator, built entirely in PyTorch, primarily intende...
5 versions - Latest release: over 1 year ago - 1 dependent package - 86 downloads last month - 38 stars on GitHub - 2 maintainers
Top 5.9% on pypi.org
carvekit 4.1.2
Open-Source background removal framework
11 versions - Latest release: over 1 year ago - 1 dependent package - 2 dependent repositories - 1.76 thousand downloads last month - 1,517 stars on GitHub - 1 maintainer
dtrc 0.0.15
data torch
14 versions - Latest release: over 1 year ago - 100 downloads last month - 1 maintainer
neuroimage-denoiser-gui 1.4.4
GUI Implementation for Neuorimage Denoiser (https://github.com/s-weissbach/neuroimage_denoiser)
8 versions - Latest release: 7 months ago - 31 downloads last month - 0 stars on GitHub - 1 maintainer
visualtorch 0.2.4
Architecture visualization of Torch models
8 versions - Latest release: 4 months ago - 771 downloads last month - 20 stars on GitHub - 1 maintainer
phaseaug 1.0.1
PhaseAug: A Differentiable Augmentation for Speech Synthesis to Simulate One-to-Many Mapping
3 versions - Latest release: over 2 years ago - 1 dependent package - 1 dependent repositories - 1.89 thousand downloads last month - 189 stars on GitHub - 1 maintainer
maestro 1.0.0
Streamline the fine-tuning process for vision-language models like PaliGemma 2, Florence-2, and Q...
17 versions - Latest release: 9 months ago - 3 dependent repositories - 599 downloads last month - 1,036 stars on GitHub - 3 maintainers
matrepr 1.0.1
Format matrices and tensors to HTML, string, and LaTeX, with Jupyter integration.
16 versions - Latest release: over 1 year ago - 2 dependent packages - 250 downloads last month - 13 stars on GitHub - 1 maintainer
nemo-aligner 0.7.0
NeMo-Aligner - a toolkit for model alignment
10 versions - Latest release: 8 months ago - 79 downloads last month - 841 stars on GitHub - 3 maintainers
tilus 0.1.1
Tilus
3 versions - Latest release: 2 months ago - 58 downloads last month - 386 stars on GitHub
rlbidder 0.1.0
A modern, modular RL library for auto-bidding in online advertising auctions.
1 version - Latest release: 26 days ago - 139 downloads last month - 1 stars on GitHub - 1 maintainer
chemicalx 0.1.0
A Deep Learning Library for Drug Pair Scoring.
10 versions - Latest release: over 3 years ago - 1 dependent repositories - 116 downloads last month - 748 stars on GitHub - 3 maintainers
Top 5.8% on pypi.org
torchkeras 4.0.5
pytorch❤️keras
59 versions - Latest release: about 1 month ago - 4 dependent repositories - 1.02 thousand downloads last month - 1,980 stars on GitHub - 1 maintainer
stable-fast-pruna 1.0.7 💰
Stable Fast is an ultra lightweight performance optimization framework for Hugging Fase diffuser ...
3 versions - Latest release: 6 months ago - 1.29 thousand downloads last month - 1,286 stars on GitHub - 1 maintainer
torchfunc-nightly 1663034600
PyTorch functions to improve performance, analyse models and make your life easier.
1,092 versions - Latest release: about 3 years ago - 1 dependent repositories - 2.56 thousand downloads last month - 195 stars on GitHub - 1 maintainer
fastner 0.1.3
Finetune transformer-based models for the Named Entity Recognition task in a simple and fast way.
21 versions - Latest release: about 3 years ago - 17 downloads last month - 1 stars on GitHub - 1 maintainer
shiva-si 0.1
Dataset-agnostic model saving and loading format (.si) for AI/ML pipelines
1 version - Latest release: 4 months ago - 10 downloads last month - 0 stars on GitHub - 1 maintainer
Top 9.1% on pypi.org
torchrl-nightly 2025.9.1
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning
929 versions - Latest release: 2 months ago - 63.7 thousand downloads last month - 1,571 stars on GitHub - 1 maintainer
glasflow 0.4.1
Normalising flows using nflows
7 versions - Latest release: about 1 year ago - 3 dependent packages - 1 dependent repositories - 14.2 thousand downloads last month - 27 stars on GitHub - 2 maintainers
pynmsnn 0.0.3
NeuroMorphic Predictive Model with Spiking Neural Networks (SNN) using Pytorch
3 versions - Latest release: over 4 years ago - 1 dependent repositories - 26 downloads last month - 32 stars on GitHub - 1 maintainer
Top 3.3% on pypi.org
torch-tensorrt 2.9.0
Torch-TensorRT is a package which allows users to automatically compile PyTorch and TorchScript m...
13 versions - Latest release: 17 days ago - 3 dependent packages - 4 dependent repositories - 57.7 thousand downloads last month - 2,874 stars on GitHub - 3 maintainers
keyedtensor 0.3.0
keyed torch tensors
4 versions - Latest release: about 5 years ago - 1 dependent repositories - 15 downloads last month - 0 stars on GitHub - 1 maintainer
redoc 0.2.3
Universal document converter with template support, OCR, and AI-powered processing. Convert betwe...
4 versions - Latest release: 5 months ago - 377 downloads last month - 0 stars on GitHub - 1 maintainer
torchsparsegradutils 0.2.1
A collection of utility functions to work with PyTorch sparse tensors
6 versions - Latest release: 5 days ago - 1 dependent repositories - 276 downloads last month - 32 stars on GitHub - 1 maintainer
torchhull 0.2.0
A fast Visual Hull implementation
2 versions - Latest release: 4 months ago - 40 downloads last month - 11 stars on GitHub - 1 maintainer
autofrk 1.2.1
autoFRK: Automatic Fixed Rank Kriging. The Python version with PyTorch
5 versions - Latest release: 5 days ago - 419 downloads last month - 1 stars on GitHub - 1 maintainer
Top 3.6% on pypi.org
torchtnt 0.2.4
A lightweight library for PyTorch training tools and utilities
13 versions - Latest release: over 1 year ago - 2 dependent packages - 5 dependent repositories - 130 thousand downloads last month - 1,712 stars on GitHub - 2 maintainers
potc-torch 0.0.1
Torch support of potc.
1 version - Latest release: almost 4 years ago - 1 dependent package - 1 dependent repositories - 13 downloads last month - 0 stars on GitHub - 1 maintainer
Top 7.8% on pypi.org
audtorch 0.6.4
Deep learning with PyTorch and audio
15 versions - Latest release: about 5 years ago - 1 dependent package - 6 dependent repositories - 1.63 thousand downloads last month - 86 stars on GitHub - 1 maintainer
omninumpy 0.1.65
Compatibility layer for NumPy 2.x with GPU acceleration support
6 versions - Latest release: 23 days ago - 660 downloads last month - 1 stars on GitHub - 1 maintainer
torchruntime 1.24.0
Meant for app developers. A convenient way to install and configure the appropriate version of Py...
51 versions - Latest release: about 1 month ago - 24.6 thousand downloads last month - 14 stars on GitHub - 1 maintainer
torchxlstm 0.0.3
A pure pytorch implementation of xLSTM.
2 versions - Latest release: over 1 year ago - 61 downloads last month - 52 stars on GitHub - 1 maintainer
megatron-bridge 0.1.0rc4
Megatron Bridge: Training Recipes for Megatron-based LLM and VLM models
14 versions - Latest release: 11 days ago - 3.64 thousand downloads last month - 120 stars on GitHub - 2 maintainers
Top 4.5% on pypi.org
bittensor 9.12.1
Bittensor
173 versions - Latest release: 14 days ago - 6 dependent packages - 2 dependent repositories - 147 thousand downloads last month - 1,279 stars on GitHub - 3 maintainers
histomics-stream 2.5.3
Whole-slide image streamer for machine learning frameworks.
23 versions - Latest release: about 1 year ago - 1 dependent repositories - 72 downloads last month - 24 stars on GitHub - 1 maintainer
diffusionlm 0.1.9 💰
A diffusion-based language model implementation
7 versions - Latest release: 7 months ago - 67 downloads last month - 4 stars on GitHub - 1 maintainer
quantllm 1.0.0 💰
A lightweight library for quantized LLM fine-tuning and deployment
7 versions - Latest release: 6 months ago - 26 downloads last month - 3 stars on GitHub - 1 maintainer
Top 3.6% on pypi.org
torchfile 0.1.0
Torch7 binary serialized file parser
2 versions - Latest release: over 8 years ago - 2 dependent packages - 695 dependent repositories - 60 thousand downloads last month - 211 stars on GitHub - 1 maintainer
torchlens 0.1.36
A package for extracting activations from PyTorch models
36 versions - Latest release: about 1 month ago - 1 dependent repositories - 536 downloads last month - 620 stars on GitHub - 1 maintainer
owlite 0.0.8
A fake package to warn the user they are not installing the correct package.
6 versions - Latest release: about 1 year ago - 13 downloads last month - 49 stars on GitHub - 2 maintainers
statictorch 0.0.4
Provides type annotations for torch tensors.
4 versions - Latest release: 8 months ago - 16 downloads last month - 0 stars on GitHub - 1 maintainer
qmb 0.0.52
Quantum Manybody Problem
9 versions - Latest release: about 2 months ago - 33 downloads last month - 0 stars on GitHub - 1 maintainer
kvmm 0.1.8
Pretrained keras 3 vision models
8 versions - Latest release: 3 months ago - 43 downloads last month - 8 stars on GitHub - 1 maintainer
sequifier 3.0.0
Train a transformer model with the command line
78 versions - Latest release: over 1 year ago - 706 downloads last month - 5 stars on GitHub - 1 maintainer
torchspd 0.1.2
Differentiable spectral operators on symmetric positive definite (SPD) matrices in PyTorch
2 versions - Latest release: about 2 months ago - 20 downloads last month - 0 stars on GitHub - 1 maintainer
Top 6.9% on pypi.org
piqa 1.3.2
PyTorch Image Quality Assessment
15 versions - Latest release: about 2 years ago - 7 dependent repositories - 6 thousand downloads last month - 425 stars on GitHub - 1 maintainer
Top 3.9% on pypi.org
rtdl 0.0.13
Research on Tabular Deep Learning.
18 versions - Latest release: over 3 years ago - 3 dependent packages - 9 dependent repositories - 8.27 thousand downloads last month - 1,041 stars on GitHub - 1 maintainer
Top 6.7% on pypi.org
diffdist 0.1
Provides differentiable communication primitives such as Send/Recv/Gather etc for pytorch. Autogr...
1 version - Latest release: over 6 years ago - 78 dependent repositories - 5.14 thousand downloads last month - 62 stars on GitHub - 1 maintainer
openvalidators 1.0.4
Openvalidators is a collection of open source validators for the Bittensor Network.
5 versions - Latest release: over 2 years ago - 20 downloads last month - 15 stars on GitHub - 1 maintainer
Top 6.8% on pypi.org
concrete-ml 1.9.0
Concrete ML is an open-source set of tools which aims to simplify the use of fully homomorphic en...
31 versions - Latest release: 7 months ago - 2 dependent repositories - 2.06 thousand downloads last month - 584 stars on GitHub - 1 maintainer
nclustrl 1.0.0
Toolbox to learn biclustering and triclustering task using Ray's rllib and torch
1 version - Latest release: over 3 years ago - 1 dependent repositories - 11 downloads last month - 0 stars on GitHub - 1 maintainer