Ecosyste.ms: Packages
An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.
pypi.org "pytorch" keyword
Top 0.1% on pypi.org
36 versions - Latest release: 17 days ago - 5,049 dependent packages - 92,053 dependent repositories - 28.3 million downloads last month - 77,536 stars on GitHub - 7 maintainers
torch 2.3.0
Tensors and Dynamic neural networks in Python with strong GPU acceleration36 versions - Latest release: 17 days ago - 5,049 dependent packages - 92,053 dependent repositories - 28.3 million downloads last month - 77,536 stars on GitHub - 7 maintainers
Top 0.1% on pypi.org
141 versions - Latest release: 5 days ago - 1,523 dependent packages - 31,800 dependent repositories - 27.1 million downloads last month - 125,725 stars on GitHub - 6 maintainers
transformers 4.40.2
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow141 versions - Latest release: 5 days ago - 1,523 dependent packages - 31,800 dependent repositories - 27.1 million downloads last month - 125,725 stars on GitHub - 6 maintainers
Top 0.1% on pypi.org
99 versions - Latest release: 15 days ago - 392 dependent packages - 24,558 dependent repositories - 18 million downloads last month - 60,873 stars on GitHub - 2 maintainers
keras 3.3.3
Multi-backend Keras.99 versions - Latest release: 15 days ago - 392 dependent packages - 24,558 dependent repositories - 18 million downloads last month - 60,873 stars on GitHub - 2 maintainers
Top 1.1% on pypi.org
109 versions - Latest release: 9 days ago - 430 dependent packages - 15,445 dependent repositories - 33.2 million downloads last month - 1,641 stars on GitHub - 5 maintainers
huggingface-hub 0.23.0
Client library to download and publish models, datasets and other repos on the huggingface.co hub109 versions - Latest release: 9 days ago - 430 dependent packages - 15,445 dependent repositories - 33.2 million downloads last month - 1,641 stars on GitHub - 5 maintainers
Top 0.2% on pypi.org
83 versions - Latest release: 5 days ago - 650 dependent packages - 14,962 dependent repositories - 9.46 million downloads last month - 18,474 stars on GitHub - 4 maintainers
datasets 2.19.1
HuggingFace community-driven open-source library of datasets83 versions - Latest release: 5 days ago - 650 dependent packages - 14,962 dependent repositories - 9.46 million downloads last month - 18,474 stars on GitHub - 4 maintainers
Top 0.8% on pypi.org
15 versions - Latest release: 13 days ago - 587 dependent packages - 14,487 dependent repositories - 6.93 million downloads last month - 7,927 stars on GitHub - 1 maintainer
einops 0.8.0
A new flavour of deep learning operations15 versions - Latest release: 13 days ago - 587 dependent packages - 14,487 dependent repositories - 6.93 million downloads last month - 7,927 stars on GitHub - 1 maintainer
Top 0.8% on pypi.org
31 versions - Latest release: 17 days ago - 361 dependent packages - 11,395 dependent repositories - 6.1 million downloads last month - 2,382 stars on GitHub - 8 maintainers
torchaudio 2.3.0
An audio package for PyTorch31 versions - Latest release: 17 days ago - 361 dependent packages - 11,395 dependent repositories - 6.1 million downloads last month - 2,382 stars on GitHub - 8 maintainers
Top 0.2% on pypi.org
38 versions - Latest release: 29 days ago - 472 dependent packages - 9,609 dependent repositories - 6.39 million downloads last month - 12,567 stars on GitHub - 1 maintainer
onnxruntime 1.17.3
ONNX Runtime is a runtime accelerator for Machine Learning models38 versions - Latest release: 29 days ago - 472 dependent packages - 9,609 dependent repositories - 6.39 million downloads last month - 12,567 stars on GitHub - 1 maintainer
Top 0.7% on pypi.org
266 versions - Latest release: 4 days ago - 401 dependent packages - 9,299 dependent repositories - 12.8 million downloads last month - 8,194 stars on GitHub - 8 maintainers
wandb 0.17.0
A CLI and library for interacting with the Weights & Biases API.266 versions - Latest release: 4 days ago - 401 dependent packages - 9,299 dependent repositories - 12.8 million downloads last month - 8,194 stars on GitHub - 8 maintainers
Top 0.2% on pypi.org
196 versions - Latest release: 10 days ago - 477 dependent packages - 9,165 dependent repositories - 5.08 million downloads last month - 26,951 stars on GitHub - 2 maintainers
pytorch-lightning 2.2.4
PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write...196 versions - Latest release: 10 days ago - 477 dependent packages - 9,165 dependent repositories - 5.08 million downloads last month - 26,951 stars on GitHub - 2 maintainers
Top 0.2% on pypi.org
31 versions - Latest release: about 2 months ago - 369 dependent packages - 8,716 dependent repositories - 5.59 million downloads last month - 16,165 stars on GitHub - 4 maintainers
onnx 1.16.0
Open Neural Network Exchange31 versions - Latest release: about 2 months ago - 369 dependent packages - 8,716 dependent repositories - 5.59 million downloads last month - 16,165 stars on GitHub - 4 maintainers
Top 0.9% on pypi.org
232 versions - Latest release: 22 days ago - 28 dependent packages - 8,632 dependent repositories - 9.08 million downloads last month - 2,792 stars on GitHub - 3 maintainers
thinc 9.0.0
A refreshing functional take on deep learning, compatible with your favorite libraries232 versions - Latest release: 22 days ago - 28 dependent packages - 8,632 dependent repositories - 9.08 million downloads last month - 2,792 stars on GitHub - 3 maintainers
Top 0.2% on pypi.org
56 versions - Latest release: 3 months ago - 250 dependent packages - 7,759 dependent repositories - 5.14 million downloads last month - 29,904 stars on GitHub - 1 maintainer
timm 0.9.16 💰
PyTorch Image Models56 versions - Latest release: 3 months ago - 250 dependent packages - 7,759 dependent repositories - 5.14 million downloads last month - 29,904 stars on GitHub - 1 maintainer
Top 0.9% on pypi.org
53 versions - Latest release: 5 days ago - 263 dependent packages - 7,449 dependent repositories - 4.68 million downloads last month - 1,953 stars on GitHub - 2 maintainers
torchmetrics 1.4.0
PyTorch native Metrics53 versions - Latest release: 5 days ago - 263 dependent packages - 7,449 dependent repositories - 4.68 million downloads last month - 1,953 stars on GitHub - 2 maintainers
Top 0.6% on pypi.org
38 versions - Latest release: about 2 months ago - 82 dependent packages - 5,630 dependent repositories - 1.49 million downloads last month - 8,984 stars on GitHub - 2 maintainers
kornia 0.7.2 💰
Open Source Differentiable Computer Vision Library for PyTorch38 versions - Latest release: about 2 months ago - 82 dependent packages - 5,630 dependent repositories - 1.49 million downloads last month - 8,984 stars on GitHub - 2 maintainers
Top 0.4% on pypi.org
74 versions - Latest release: about 2 months ago - 119 dependent packages - 5,162 dependent repositories - 2.41 million downloads last month - 20,155 stars on GitHub - 4 maintainers
diffusers 0.27.2
State-of-the-art diffusion in PyTorch and JAX.74 versions - Latest release: about 2 months ago - 119 dependent packages - 5,162 dependent repositories - 2.41 million downloads last month - 20,155 stars on GitHub - 4 maintainers
Top 0.3% on pypi.org
90 versions - Latest release: 18 days ago - 88 dependent packages - 3,724 dependent repositories - 501 thousand downloads last month - 32,440 stars on GitHub - 3 maintainers
deepspeed 0.14.2
DeepSpeed library90 versions - Latest release: 18 days ago - 88 dependent packages - 3,724 dependent repositories - 501 thousand downloads last month - 32,440 stars on GitHub - 3 maintainers
Top 0.2% on pypi.org
102 versions - Latest release: 3 days ago - 234 dependent packages - 3,641 dependent repositories - 4.58 million downloads last month - 30,888 stars on GitHub - 33 maintainers
ray 2.21.0
Ray provides a simple, universal API for building distributed applications.102 versions - Latest release: 3 days ago - 234 dependent packages - 3,641 dependent repositories - 4.58 million downloads last month - 30,888 stars on GitHub - 33 maintainers
Top 0.7% on pypi.org
33 versions - Latest release: 17 days ago - 69 dependent packages - 2,976 dependent repositories - 673 thousand downloads last month - 3,447 stars on GitHub - 6 maintainers
torchtext 0.18.0
Text utilities, models, transforms, and datasets for PyTorch.33 versions - Latest release: 17 days ago - 69 dependent packages - 2,976 dependent repositories - 673 thousand downloads last month - 3,447 stars on GitHub - 6 maintainers
Top 0.4% on pypi.org
137 versions - Latest release: 14 days ago - 50 dependent packages - 2,806 dependent repositories - 373 thousand downloads last month - 25,661 stars on GitHub - 2 maintainers
fastai 2.7.15
fastai simplifies training fast and accurate neural nets using modern best practices137 versions - Latest release: 14 days ago - 50 dependent packages - 2,806 dependent repositories - 373 thousand downloads last month - 25,661 stars on GitHub - 2 maintainers
Top 1.2% on pypi.org
7 versions - Latest release: over 5 years ago - 39 dependent packages - 2,252 dependent repositories - 262 thousand downloads last month - 3,955 stars on GitHub - 2 maintainers
torchsummary 1.5.1
Model summary in PyTorch similar to `model.summary()` in Keras7 versions - Latest release: over 5 years ago - 39 dependent packages - 2,252 dependent repositories - 262 thousand downloads last month - 3,955 stars on GitHub - 2 maintainers
Top 0.8% on pypi.org
24 versions - Latest release: 10 months ago - 63 dependent packages - 2,224 dependent repositories - 2.18 million downloads last month - 7,802 stars on GitHub - 1 maintainer
tensorboardx 2.6.2
TensorBoardX lets you watch Tensors Flow without Tensorflow24 versions - Latest release: 10 months ago - 63 dependent packages - 2,224 dependent repositories - 2.18 million downloads last month - 7,802 stars on GitHub - 1 maintainer
Top 0.6% on pypi.org
16 versions - Latest release: 4 months ago - 102 dependent packages - 2,080 dependent repositories - 681 thousand downloads last month - 10,420 stars on GitHub - 1 maintainer
open3d 0.18.0
Open3D: A Modern Library for 3D Data Processing.16 versions - Latest release: 4 months ago - 102 dependent packages - 2,080 dependent repositories - 681 thousand downloads last month - 10,420 stars on GitHub - 1 maintainer
Top 0.5% on pypi.org
11 versions - Latest release: over 1 year ago - 11 dependent packages - 2,022 dependent repositories - 577 thousand downloads last month - 34,598 stars on GitHub - 2 maintainers
gfpgan 1.3.8
GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration11 versions - Latest release: over 1 year ago - 11 dependent packages - 2,022 dependent repositories - 577 thousand downloads last month - 34,598 stars on GitHub - 2 maintainers
Top 0.7% on pypi.org
26 versions - Latest release: over 1 year ago - 13 dependent packages - 1,996 dependent repositories - 592 thousand downloads last month - 6,205 stars on GitHub - 1 maintainer
basicsr 1.4.2
Open Source Image and Video Super-Resolution Toolbox26 versions - Latest release: over 1 year ago - 13 dependent packages - 1,996 dependent repositories - 592 thousand downloads last month - 6,205 stars on GitHub - 1 maintainer
Top 1.5% on pypi.org
71 versions - Latest release: 5 months ago - 32 dependent packages - 1,774 dependent repositories - 1.65 million downloads last month - 1,984 stars on GitHub - 1 maintainer
webdataset 0.2.86
Record sequential storage for deep learning.71 versions - Latest release: 5 months ago - 32 dependent packages - 1,774 dependent repositories - 1.65 million downloads last month - 1,984 stars on GitHub - 1 maintainer
Top 1.9% on pypi.org
17 versions - Latest release: over 4 years ago - 16 dependent packages - 1,658 dependent repositories - 60.6 thousand downloads last month - 735 stars on GitHub - 2 maintainers
test-tube 0.7.5
Experiment logger and visualizer17 versions - Latest release: over 4 years ago - 16 dependent packages - 1,658 dependent repositories - 60.6 thousand downloads last month - 735 stars on GitHub - 2 maintainers
Top 1.1% on pypi.org
5 versions - Latest release: over 2 years ago - 28 dependent packages - 1,578 dependent repositories - 494 thousand downloads last month - 3,376 stars on GitHub - 1 maintainer
lpips 0.1.4
LPIPS Similarity metric5 versions - Latest release: over 2 years ago - 28 dependent packages - 1,578 dependent repositories - 494 thousand downloads last month - 3,376 stars on GitHub - 1 maintainer
Top 0.5% on pypi.org
14 versions - Latest release: over 1 year ago - 16 dependent packages - 1,562 dependent repositories - 427 thousand downloads last month - 25,959 stars on GitHub - 1 maintainer
realesrgan 0.3.0
Real-ESRGAN aims at developing Practical Algorithms for General Image Restoration14 versions - Latest release: over 1 year ago - 16 dependent packages - 1,562 dependent repositories - 427 thousand downloads last month - 25,959 stars on GitHub - 1 maintainer
Top 1.6% on pypi.org
740 versions - Latest release: 5 months ago - 29 dependent packages - 1,390 dependent repositories - 73.9 thousand downloads last month - 2,765 stars on GitHub - 1 maintainer
docarray 0.40.0
The data structure for multimodal data740 versions - Latest release: 5 months ago - 29 dependent packages - 1,390 dependent repositories - 73.9 thousand downloads last month - 2,765 stars on GitHub - 1 maintainer
Top 1.3% on pypi.org
141 versions - Latest release: about 18 hours ago - 27 dependent packages - 1,384 dependent repositories - 52.6 thousand downloads last month - 7,733 stars on GitHub - 5 maintainers
deeplake 3.9.5
Activeloop Deep Lake141 versions - Latest release: about 18 hours ago - 27 dependent packages - 1,384 dependent repositories - 52.6 thousand downloads last month - 7,733 stars on GitHub - 5 maintainers
Top 1.6% on pypi.org
2 versions - Latest release: 8 months ago - 9 dependent packages - 1,277 dependent repositories - 1.41 million downloads last month - 1,470 stars on GitHub - 4 maintainers
torchsde 0.2.6
SDE solvers and stochastic adjoint sensitivity analysis in PyTorch.2 versions - Latest release: 8 months ago - 9 dependent packages - 1,277 dependent repositories - 1.41 million downloads last month - 1,470 stars on GitHub - 4 maintainers
Top 0.4% on pypi.org
37 versions - Latest release: 3 months ago - 89 dependent packages - 1,276 dependent repositories - 1.38 million downloads last month - 12,567 stars on GitHub - 1 maintainer
onnxruntime-gpu 1.17.1
ONNX Runtime is a runtime accelerator for Machine Learning models37 versions - Latest release: 3 months ago - 89 dependent packages - 1,276 dependent repositories - 1.38 million downloads last month - 12,567 stars on GitHub - 1 maintainer
Top 0.5% on pypi.org
333 versions - Latest release: about 20 hours ago - 68 dependent packages - 1,176 dependent repositories - 2.88 million downloads last month - 17,533 stars on GitHub - 1 maintainer
ultralytics 8.2.12 💰
Ultralytics YOLOv8 for SOTA object detection, multi-object tracking, instance segmentation, pose ...333 versions - Latest release: about 20 hours ago - 68 dependent packages - 1,176 dependent repositories - 2.88 million downloads last month - 17,533 stars on GitHub - 1 maintainer
Top 0.7% on pypi.org
43 versions - Latest release: 4 months ago - 58 dependent packages - 1,154 dependent repositories - 2.11 million downloads last month - 8,566 stars on GitHub - 2 maintainers
open-clip-torch 2.24.0
OpenCLIP43 versions - Latest release: 4 months ago - 58 dependent packages - 1,154 dependent repositories - 2.11 million downloads last month - 8,566 stars on GitHub - 2 maintainers
Top 0.7% on pypi.org
265 versions - Latest release: over 1 year ago - 23 dependent packages - 1,128 dependent repositories - 50.7 thousand downloads last month - 11,693 stars on GitHub - 3 maintainers
allennlp 2.10.1
An open-source NLP research library, built on PyTorch.265 versions - Latest release: over 1 year ago - 23 dependent packages - 1,128 dependent repositories - 50.7 thousand downloads last month - 11,693 stars on GitHub - 3 maintainers
Top 1.4% on pypi.org
23 versions - Latest release: 7 months ago - 40 dependent packages - 1,102 dependent repositories - 38.7 thousand downloads last month - 1,460 stars on GitHub - 1 maintainer
torch-scatter 2.1.2
PyTorch Extension Library of Optimized Scatter Operations23 versions - Latest release: 7 months ago - 40 dependent packages - 1,102 dependent repositories - 38.7 thousand downloads last month - 1,460 stars on GitHub - 1 maintainer
Top 0.5% on pypi.org
42 versions - Latest release: 22 days ago - 95 dependent packages - 1,062 dependent repositories - 285 thousand downloads last month - 19,882 stars on GitHub - 1 maintainer
torch-geometric 2.5.3
Graph Neural Network Library for PyTorch42 versions - Latest release: 22 days ago - 95 dependent packages - 1,062 dependent repositories - 285 thousand downloads last month - 19,882 stars on GitHub - 1 maintainer
Top 2.0% on pypi.org
35 versions - Latest release: over 1 year ago - 3 dependent packages - 1,060 dependent repositories - 798 thousand downloads last month - 855 stars on GitHub - 1 maintainer
clean-fid 0.1.35
FID calculation in PyTorch with proper image resizing and quantization steps35 versions - Latest release: over 1 year ago - 3 dependent packages - 1,060 dependent repositories - 798 thousand downloads last month - 855 stars on GitHub - 1 maintainer
Top 0.5% on pypi.org
1,083 versions - Latest release: 26 days ago - 17 dependent packages - 1,025 dependent repositories - 339 thousand downloads last month - 60,873 stars on GitHub - 3 maintainers
keras-nightly 3.2.1.dev2024041504
Multi-backend Keras.1,083 versions - Latest release: 26 days ago - 17 dependent packages - 1,025 dependent repositories - 339 thousand downloads last month - 60,873 stars on GitHub - 3 maintainers
Top 0.6% on pypi.org
10 versions - Latest release: about 5 years ago - 9 dependent packages - 940 dependent repositories - 25.4 thousand downloads last month - 125,524 stars on GitHub - 1 maintainer
pytorch-pretrained-bert 0.6.2
PyTorch version of Google AI BERT model with script to load Google pre-trained models10 versions - Latest release: about 5 years ago - 9 dependent packages - 940 dependent repositories - 25.4 thousand downloads last month - 125,524 stars on GitHub - 1 maintainer
Top 0.4% on pypi.org
72 versions - Latest release: over 1 year ago - 120 dependent packages - 903 dependent repositories - 897 thousand downloads last month - 26,951 stars on GitHub - 4 maintainers
lightning 2022.10.25
The Deep Learning framework to train, deploy, and ship AI products Lightning fast.72 versions - Latest release: over 1 year ago - 120 dependent packages - 903 dependent repositories - 897 thousand downloads last month - 26,951 stars on GitHub - 4 maintainers
Top 1.0% on pypi.org
38 versions - Latest release: over 1 year ago - 51 dependent packages - 879 dependent repositories - 186 thousand downloads last month - 3,419 stars on GitHub - 2 maintainers
gpytorch 1.9.1
An implementation of Gaussian Processes in Pytorch38 versions - Latest release: over 1 year ago - 51 dependent packages - 879 dependent repositories - 186 thousand downloads last month - 3,419 stars on GitHub - 2 maintainers
Top 1.9% on pypi.org
7 versions - Latest release: about 1 year ago - 16 dependent packages - 854 dependent repositories - 9.86 thousand downloads last month - 1,369 stars on GitHub - 1 maintainer
functorch 2.0.0
JAX-like composable function transforms for PyTorch7 versions - Latest release: about 1 year ago - 16 dependent packages - 854 dependent repositories - 9.86 thousand downloads last month - 1,369 stars on GitHub - 1 maintainer
Top 1.6% on pypi.org
16 versions - Latest release: about 1 year ago - 13 dependent packages - 780 dependent repositories - 978 thousand downloads last month - 745 stars on GitHub - 2 maintainers
facexlib 0.3.0
Basic face library16 versions - Latest release: about 1 year ago - 13 dependent packages - 780 dependent repositories - 978 thousand downloads last month - 745 stars on GitHub - 2 maintainers
Top 2.2% on pypi.org
3 versions - Latest release: almost 3 years ago - 6 dependent packages - 773 dependent repositories - 282 thousand downloads last month - 870 stars on GitHub - 2 maintainers
torch-fidelity 0.3.0
High-fidelity performance metrics for generative models in PyTorch3 versions - Latest release: almost 3 years ago - 6 dependent packages - 773 dependent repositories - 282 thousand downloads last month - 870 stars on GitHub - 2 maintainers
Top 0.7% on pypi.org
4 versions - Latest release: over 4 years ago - 12 dependent packages - 772 dependent repositories - 48.5 thousand downloads last month - 123,549 stars on GitHub - 1 maintainer
pytorch-transformers 1.2.0
Repository of pre-trained NLP Transformer models: BERT & RoBERTa, GPT & GPT-2, Transformer-XL, XL...4 versions - Latest release: over 4 years ago - 12 dependent packages - 772 dependent repositories - 48.5 thousand downloads last month - 123,549 stars on GitHub - 1 maintainer
Top 1.0% on pypi.org
1,563 versions - Latest release: 7 months ago - 37 dependent packages - 766 dependent repositories - 238 thousand downloads last month - 4,435 stars on GitHub - 3 maintainers
pytorch-ignite 0.4.13 💰
A lightweight library to help with training neural networks in PyTorch.1,563 versions - Latest release: 7 months ago - 37 dependent packages - 766 dependent repositories - 238 thousand downloads last month - 4,435 stars on GitHub - 3 maintainers
Top 0.5% on pypi.org
13 versions - Latest release: almost 2 years ago - 32 dependent packages - 759 dependent repositories - 299 thousand downloads last month - 28,626 stars on GitHub - 4 maintainers
fairseq 0.12.2
Facebook AI Research Sequence-to-Sequence Toolkit13 versions - Latest release: almost 2 years ago - 32 dependent packages - 759 dependent repositories - 299 thousand downloads last month - 28,626 stars on GitHub - 4 maintainers
Top 1.4% on pypi.org
63 versions - Latest release: 2 days ago - 51 dependent packages - 715 dependent repositories - 1.12 million downloads last month - 2,003 stars on GitHub - 8 maintainers
optimum 1.19.2
Optimum Library is an extension of the Hugging Face Transformers library, providing a framework t...63 versions - Latest release: 2 days ago - 51 dependent packages - 715 dependent repositories - 1.12 million downloads last month - 2,003 stars on GitHub - 8 maintainers
Top 1.1% on pypi.org
47 versions - Latest release: about 1 year ago - 30 dependent packages - 688 dependent repositories - 270 thousand downloads last month - 3,837 stars on GitHub - 3 maintainers
coremltools 6.3.0
Community Tools for Core ML47 versions - Latest release: about 1 year ago - 30 dependent packages - 688 dependent repositories - 270 thousand downloads last month - 3,837 stars on GitHub - 3 maintainers
Top 0.9% on pypi.org
583 versions - Latest release: 3 days ago - 55 dependent packages - 687 dependent repositories - 26.9 million downloads last month - 2,043 stars on GitHub - 1 maintainer
sagemaker 2.219.0
Open source library for training and deploying models on Amazon SageMaker.583 versions - Latest release: 3 days ago - 55 dependent packages - 687 dependent repositories - 26.9 million downloads last month - 2,043 stars on GitHub - 1 maintainer
Top 0.7% on pypi.org
32 versions - Latest release: 8 months ago - 49 dependent packages - 671 dependent repositories - 225 thousand downloads last month - 22,043 stars on GitHub - 1 maintainer
easyocr 1.7.1
End-to-End Multi-Lingual Optical Character Recognition (OCR) Solution32 versions - Latest release: 8 months ago - 49 dependent packages - 671 dependent repositories - 225 thousand downloads last month - 22,043 stars on GitHub - 1 maintainer
Top 0.7% on pypi.org
53 versions - Latest release: 4 months ago - 29 dependent packages - 645 dependent repositories - 208 thousand downloads last month - 27,836 stars on GitHub - 2 maintainers
mmdet 3.3.0
OpenMMLab Detection Toolbox and Benchmark53 versions - Latest release: 4 months ago - 29 dependent packages - 645 dependent repositories - 208 thousand downloads last month - 27,836 stars on GitHub - 2 maintainers
Top 0.9% on pypi.org
87 versions - Latest release: 14 days ago - 59 dependent packages - 641 dependent repositories - 145 thousand downloads last month - 7,977 stars on GitHub - 5 maintainers
stable-baselines3 2.3.2
Pytorch version of Stable Baselines, implementations of reinforcement learning algorithms.87 versions - Latest release: 14 days ago - 59 dependent packages - 641 dependent repositories - 145 thousand downloads last month - 7,977 stars on GitHub - 5 maintainers
Top 2.6% on pypi.org
22 versions - Latest release: 7 months ago - 10 dependent packages - 596 dependent repositories - 193 thousand downloads last month - 350 stars on GitHub - 2 maintainers
torchcrepe 0.0.22
Pytorch implementation of CREPE pitch tracker22 versions - Latest release: 7 months ago - 10 dependent packages - 596 dependent repositories - 193 thousand downloads last month - 350 stars on GitHub - 2 maintainers
Top 1.6% on pypi.org
30 versions - Latest release: 7 months ago - 41 dependent packages - 579 dependent repositories - 26.7 thousand downloads last month - 939 stars on GitHub - 1 maintainer
torch-sparse 0.6.18
PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations30 versions - Latest release: 7 months ago - 41 dependent packages - 579 dependent repositories - 26.7 thousand downloads last month - 939 stars on GitHub - 1 maintainer
Top 0.8% on pypi.org
16 versions - Latest release: over 5 years ago - 13 dependent packages - 545 dependent repositories - 112 thousand downloads last month - 8,944 stars on GitHub - 1 maintainer
pretrainedmodels 0.7.4
Pretrained models for Pytorch16 versions - Latest release: over 5 years ago - 13 dependent packages - 545 dependent repositories - 112 thousand downloads last month - 8,944 stars on GitHub - 1 maintainer
Top 1.2% on pypi.org
17 versions - Latest release: about 2 months ago - 71 dependent packages - 527 dependent repositories - 1.67 million downloads last month - 13,589 stars on GitHub - 8 maintainers
peft 0.10.0
Parameter-Efficient Fine-Tuning (PEFT)17 versions - Latest release: about 2 months ago - 71 dependent packages - 527 dependent repositories - 1.67 million downloads last month - 13,589 stars on GitHub - 8 maintainers
Top 0.7% on pypi.org
31 versions - Latest release: 5 months ago - 28 dependent packages - 497 dependent repositories - 178 thousand downloads last month - 13,329 stars on GitHub - 1 maintainer
flair 0.13.1 💰
A very simple framework for state-of-the-art NLP31 versions - Latest release: 5 months ago - 28 dependent packages - 497 dependent repositories - 178 thousand downloads last month - 13,329 stars on GitHub - 1 maintainer
Top 1.7% on pypi.org
32 versions - Latest release: 12 months ago - 43 dependent packages - 489 dependent repositories - 150 thousand downloads last month - 2,301 stars on GitHub - 1 maintainer
torchinfo 1.8.0 💰
Model summary in PyTorch, based off of the original torchsummary.32 versions - Latest release: 12 months ago - 43 dependent packages - 489 dependent repositories - 150 thousand downloads last month - 2,301 stars on GitHub - 1 maintainer
Top 1.7% on pypi.org
27 versions - Latest release: 18 days ago - 30 dependent packages - 454 dependent repositories - 218 thousand downloads last month - 928 stars on GitHub - 2 maintainers
mmengine 0.10.4
Engine of OpenMMLab projects27 versions - Latest release: 18 days ago - 30 dependent packages - 454 dependent repositories - 218 thousand downloads last month - 928 stars on GitHub - 2 maintainers
Top 1.0% on pypi.org
21 versions - Latest release: 21 days ago - 45 dependent packages - 450 dependent repositories - 206 thousand downloads last month - 7,037 stars on GitHub - 8 maintainers
stanza 1.8.2
A Python NLP Library for Many Human Languages, by the Stanford NLP Group21 versions - Latest release: 21 days ago - 45 dependent packages - 450 dependent repositories - 206 thousand downloads last month - 7,037 stars on GitHub - 8 maintainers
Top 3.0% on pypi.org
12 versions - Latest release: about 1 year ago - 15 dependent packages - 437 dependent repositories - 6.31 thousand downloads last month - 164 stars on GitHub - 1 maintainer
torch-spline-conv 1.2.2
Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch12 versions - Latest release: about 1 year ago - 15 dependent packages - 437 dependent repositories - 6.31 thousand downloads last month - 164 stars on GitHub - 1 maintainer
Top 0.6% on pypi.org
35 versions - Latest release: 3 months ago - 64 dependent packages - 437 dependent repositories - 269 thousand downloads last month - 8,376 stars on GitHub - 2 maintainers
pyro-ppl 1.9.0 💰
A Python library for probabilistic modeling and inference35 versions - Latest release: 3 months ago - 64 dependent packages - 437 dependent repositories - 269 thousand downloads last month - 8,376 stars on GitHub - 2 maintainers
Top 1.9% on pypi.org
34 versions - Latest release: 7 months ago - 19 dependent packages - 410 dependent repositories - 13.6 thousand downloads last month - 752 stars on GitHub - 1 maintainer
torch-cluster 1.6.3
PyTorch Extension Library of Optimized Graph Cluster Algorithms34 versions - Latest release: 7 months ago - 19 dependent packages - 410 dependent repositories - 13.6 thousand downloads last month - 752 stars on GitHub - 1 maintainer
Top 1.3% on pypi.org
80 versions - Latest release: 2 months ago - 20 dependent packages - 383 dependent repositories - 95.1 thousand downloads last month - 3,372 stars on GitHub - 2 maintainers
onnx-simplifier 0.4.36
Simplify your ONNX model80 versions - Latest release: 2 months ago - 20 dependent packages - 383 dependent repositories - 95.1 thousand downloads last month - 3,372 stars on GitHub - 2 maintainers
Top 1.9% on pypi.org
16 versions - Latest release: 7 months ago - 12 dependent packages - 379 dependent repositories - 151 thousand downloads last month - 622 stars on GitHub - 5 maintainers
torch-tb-profiler 0.4.3
PyTorch Profiler TensorBoard Plugin16 versions - Latest release: 7 months ago - 12 dependent packages - 379 dependent repositories - 151 thousand downloads last month - 622 stars on GitHub - 5 maintainers
Top 0.7% on pypi.org
73 versions - Latest release: 11 months ago - 10 dependent packages - 327 dependent repositories - 61.4 thousand downloads last month - 13,953 stars on GitHub - 2 maintainers
horovod 0.28.1
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.73 versions - Latest release: 11 months ago - 10 dependent packages - 327 dependent repositories - 61.4 thousand downloads last month - 13,953 stars on GitHub - 2 maintainers
Top 0.9% on pypi.org
13 versions - Latest release: 12 months ago - 35 dependent packages - 321 dependent repositories - 134 thousand downloads last month - 8,723 stars on GitHub - 1 maintainer
segmentation-models-pytorch 0.3.3 💰
Image segmentation models with pre-trained backbones. PyTorch.13 versions - Latest release: 12 months ago - 35 dependent packages - 321 dependent repositories - 134 thousand downloads last month - 8,723 stars on GitHub - 1 maintainer
Top 2.6% on pypi.org
11 versions - Latest release: over 1 year ago - 12 dependent packages - 308 dependent repositories - 215 thousand downloads last month - 399 stars on GitHub - 2 maintainers
julius 0.2.7
Nice DSP sweets: resampling, FFT Convolutions. All with PyTorch, differentiable and with CUDA sup...11 versions - Latest release: over 1 year ago - 12 dependent packages - 308 dependent repositories - 215 thousand downloads last month - 399 stars on GitHub - 2 maintainers
Top 1.5% on pypi.org
11 versions - Latest release: 11 months ago - 16 dependent packages - 299 dependent repositories - 39.4 thousand downloads last month - 1,645 stars on GitHub - 1 maintainer
lightning-bolts 0.7.0
Lightning Bolts is a community contribution for ML researchers.11 versions - Latest release: 11 months ago - 16 dependent packages - 299 dependent repositories - 39.4 thousand downloads last month - 1,645 stars on GitHub - 1 maintainer
Top 3.1% on pypi.org
32 versions - Latest release: 5 months ago - 3 dependent packages - 293 dependent repositories - 126 thousand downloads last month - 168 stars on GitHub - 1 maintainer
trainer 0.0.36
General purpose model trainer for PyTorch that is more flexible than it should be, by 🐸Coqui.32 versions - Latest release: 5 months ago - 3 dependent packages - 293 dependent repositories - 126 thousand downloads last month - 168 stars on GitHub - 1 maintainer
Top 1.6% on pypi.org
19 versions - Latest release: about 1 year ago - 12 dependent packages - 289 dependent repositories - 38.5 thousand downloads last month - 1,871 stars on GitHub - 1 maintainer
ogb 1.3.6
Open Graph Benchmark19 versions - Latest release: about 1 year ago - 12 dependent packages - 289 dependent repositories - 38.5 thousand downloads last month - 1,871 stars on GitHub - 1 maintainer
Top 2.3% on pypi.org
3 versions - Latest release: almost 6 years ago - 1 dependent package - 284 dependent repositories - 6.75 thousand downloads last month - 1,629 stars on GitHub - 1 maintainer
torchnet 0.0.4
an abstraction to train neural networks3 versions - Latest release: almost 6 years ago - 1 dependent package - 284 dependent repositories - 6.75 thousand downloads last month - 1,629 stars on GitHub - 1 maintainer
Top 2.0% on pypi.org
206 versions - Latest release: 2 days ago - 26 dependent packages - 273 dependent repositories - 540 thousand downloads last month - 522 stars on GitHub - 2 maintainers
neptune-client 1.10.3
Neptune Client206 versions - Latest release: 2 days ago - 26 dependent packages - 273 dependent repositories - 540 thousand downloads last month - 522 stars on GitHub - 2 maintainers
Top 1.1% on pypi.org
83 versions - Latest release: 7 months ago - 32 dependent packages - 269 dependent repositories - 155 thousand downloads last month - 5,376 stars on GitHub - 1 maintainer
monai 1.3.0
AI Toolkit for Healthcare Imaging83 versions - Latest release: 7 months ago - 32 dependent packages - 269 dependent repositories - 155 thousand downloads last month - 5,376 stars on GitHub - 1 maintainer
Top 1.3% on pypi.org
13 versions - Latest release: 9 months ago - 4 dependent packages - 256 dependent repositories - 77.9 thousand downloads last month - 6,812 stars on GitHub - 1 maintainer
face-alignment 1.4.1
Detector 2D or 3D face landmarks from Python13 versions - Latest release: 9 months ago - 4 dependent packages - 256 dependent repositories - 77.9 thousand downloads last month - 6,812 stars on GitHub - 1 maintainer
Top 1.3% on pypi.org
33 versions - Latest release: over 1 year ago - 12 dependent packages - 255 dependent repositories - 31.6 thousand downloads last month - 2,949 stars on GitHub - 1 maintainer
mmcls 0.25.0
OpenMMLab Image Classification Toolbox and Benchmark33 versions - Latest release: over 1 year ago - 12 dependent packages - 255 dependent repositories - 31.6 thousand downloads last month - 2,949 stars on GitHub - 1 maintainer
Top 1.1% on pypi.org
22 versions - Latest release: about 2 months ago - 6 dependent packages - 251 dependent repositories - 713 thousand downloads last month - 3,975 stars on GitHub - 5 maintainers
torch-model-archiver 0.10.0
Torch Model Archiver is used for creating archives of trained neural net models that can be consu...22 versions - Latest release: about 2 months ago - 6 dependent packages - 251 dependent repositories - 713 thousand downloads last month - 3,975 stars on GitHub - 5 maintainers
Top 1.0% on pypi.org
47 versions - Latest release: 5 months ago - 10 dependent packages - 249 dependent repositories - 48.8 thousand downloads last month - 7,447 stars on GitHub - 1 maintainer
mmsegmentation 1.2.2
Open MMLab Semantic Segmentation Toolbox and Benchmark47 versions - Latest release: 5 months ago - 10 dependent packages - 249 dependent repositories - 48.8 thousand downloads last month - 7,447 stars on GitHub - 1 maintainer
Top 2.4% on pypi.org
25 versions - Latest release: about 2 years ago - 4 dependent packages - 239 dependent repositories - 13.2 thousand downloads last month - 1,285 stars on GitHub - 1 maintainer
livelossplot 0.5.5 💰
Live training loss plot in Jupyter Notebook for Keras, PyTorch and others.25 versions - Latest release: about 2 years ago - 4 dependent packages - 239 dependent repositories - 13.2 thousand downloads last month - 1,285 stars on GitHub - 1 maintainer
Top 1.0% on pypi.org
108 versions - Latest release: 17 days ago - 18 dependent packages - 237 dependent repositories - 105 thousand downloads last month - 13,508 stars on GitHub - 4 maintainers
farm-haystack 1.25.5
LLM framework to build customizable, production-ready LLM applications. Connect components (model...108 versions - Latest release: 17 days ago - 18 dependent packages - 237 dependent repositories - 105 thousand downloads last month - 13,508 stars on GitHub - 4 maintainers
Top 1.6% on pypi.org
13 versions - Latest release: about 2 months ago - 6 dependent packages - 231 dependent repositories - 93.8 thousand downloads last month - 6,336 stars on GitHub - 2 maintainers
text-generation 0.7.0
Hugging Face Text Generation Python Client13 versions - Latest release: about 2 months ago - 6 dependent packages - 231 dependent repositories - 93.8 thousand downloads last month - 6,336 stars on GitHub - 2 maintainers
Top 2.6% on pypi.org
5 versions - Latest release: almost 3 years ago - 28 dependent packages - 221 dependent repositories - 37.2 thousand downloads last month - 1,335 stars on GitHub - 2 maintainers
torchtyping 0.1.4 💰
Runtime type annotations for the shape, dtype etc. of PyTorch Tensors.5 versions - Latest release: almost 3 years ago - 28 dependent packages - 221 dependent repositories - 37.2 thousand downloads last month - 1,335 stars on GitHub - 2 maintainers
Top 1.7% on pypi.org
24 versions - Latest release: 8 days ago - 17 dependent packages - 217 dependent repositories - 22.6 thousand downloads last month - 2,692 stars on GitHub - 1 maintainer
ptflops 0.7.3
Flops counter for neural networks in pytorch framework24 versions - Latest release: 8 days ago - 17 dependent packages - 217 dependent repositories - 22.6 thousand downloads last month - 2,692 stars on GitHub - 1 maintainer
Top 1.2% on pypi.org
34 versions - Latest release: over 2 years ago - 5 dependent packages - 217 dependent repositories - 73.3 thousand downloads last month - 7,825 stars on GitHub - 4 maintainers
espnet 0.10.6
ESPnet: end-to-end speech processing toolkit34 versions - Latest release: over 2 years ago - 5 dependent packages - 217 dependent repositories - 73.3 thousand downloads last month - 7,825 stars on GitHub - 4 maintainers
Top 1.5% on pypi.org
61 versions - Latest release: 17 days ago - 57 dependent packages - 214 dependent repositories - 218 thousand downloads last month - 14,037 stars on GitHub - 3 maintainers
supervision 0.20.0
A set of easy-to-use utils that will come in handy in any Computer Vision project61 versions - Latest release: 17 days ago - 57 dependent packages - 214 dependent repositories - 218 thousand downloads last month - 14,037 stars on GitHub - 3 maintainers
Top 1.7% on pypi.org
18 versions - Latest release: about 1 year ago - 23 dependent packages - 212 dependent repositories - 30.1 thousand downloads last month - 1,495 stars on GitHub - 1 maintainer
tensorly 0.8.1
Tensor learning in Python.18 versions - Latest release: about 1 year ago - 23 dependent packages - 212 dependent repositories - 30.1 thousand downloads last month - 1,495 stars on GitHub - 1 maintainer
Top 1.7% on pypi.org
3 versions - Latest release: 9 months ago - 13 dependent packages - 209 dependent repositories - 89.1 thousand downloads last month - 7,665 stars on GitHub - 1 maintainer
loralib 0.1.2
PyTorch implementation of low-rank adaptation (LoRA), a parameter-efficient approach to adapt a l...3 versions - Latest release: 9 months ago - 13 dependent packages - 209 dependent repositories - 89.1 thousand downloads last month - 7,665 stars on GitHub - 1 maintainer
Top 2.5% on pypi.org
3 versions - Latest release: almost 4 years ago - 5 dependent packages - 202 dependent repositories - 91.6 thousand downloads last month - 940 stars on GitHub - 2 maintainers
ttach 0.0.3
Images test time augmentation with PyTorch.3 versions - Latest release: almost 4 years ago - 5 dependent packages - 202 dependent repositories - 91.6 thousand downloads last month - 940 stars on GitHub - 2 maintainers
Top 1.3% on pypi.org
18 versions - Latest release: 8 months ago - 17 dependent packages - 196 dependent repositories - 163 thousand downloads last month - 5,639 stars on GitHub - 2 maintainers
skorch 0.15.0
scikit-learn compatible neural network library for pytorch18 versions - Latest release: 8 months ago - 17 dependent packages - 196 dependent repositories - 163 thousand downloads last month - 5,639 stars on GitHub - 2 maintainers
Top 1.1% on pypi.org
76 versions - Latest release: 16 days ago - 22 dependent packages - 192 dependent repositories - 230 thousand downloads last month - 1,318 stars on GitHub - 3 maintainers
spacy-transformers 1.3.5
spaCy pipelines for pre-trained BERT and other transformers76 versions - Latest release: 16 days ago - 22 dependent packages - 192 dependent repositories - 230 thousand downloads last month - 1,318 stars on GitHub - 3 maintainers
torchgen 0.0.1
Ready to use implementations of state-of-the-art generative models in PyTorch2 versions - Latest release: almost 2 years ago - 1 dependent package - 191 dependent repositories - 18.3 thousand downloads last month - 0 stars on GitHub - 1 maintainer
Top 3.3% on pypi.org
20 versions - Latest release: over 2 years ago - 182 dependent repositories - 4.04 thousand downloads last month - 3,025 stars on GitHub - 4 maintainers
kfserving 0.6.1
KFServing Python SDK20 versions - Latest release: over 2 years ago - 182 dependent repositories - 4.04 thousand downloads last month - 3,025 stars on GitHub - 4 maintainers
Top 2.1% on pypi.org
8 versions - Latest release: 12 months ago - 7 dependent packages - 182 dependent repositories - 81.9 thousand downloads last month - 968 stars on GitHub - 2 maintainers
pytorch-msssim 1.0.0
Fast and differentiable MS-SSIM and SSIM for pytorch.8 versions - Latest release: 12 months ago - 7 dependent packages - 182 dependent repositories - 81.9 thousand downloads last month - 968 stars on GitHub - 2 maintainers
Top 1.5% on pypi.org
104 versions - Latest release: about 2 years ago - 13 dependent packages - 179 dependent repositories - 12.1 thousand downloads last month - 3,230 stars on GitHub - 1 maintainer
catalyst 22.2.1 💰
Catalyst. Accelerated deep learning R&D with PyTorch.104 versions - Latest release: about 2 years ago - 13 dependent packages - 179 dependent repositories - 12.1 thousand downloads last month - 3,230 stars on GitHub - 1 maintainer
Top 1.4% on pypi.org
25 versions - Latest release: 20 days ago - 9 dependent packages - 178 dependent repositories - 46.5 thousand downloads last month - 3,025 stars on GitHub - 1 maintainer
kserve 0.12.1
KServe Python SDK25 versions - Latest release: 20 days ago - 9 dependent packages - 178 dependent repositories - 46.5 thousand downloads last month - 3,025 stars on GitHub - 1 maintainer
Top 2.1% on pypi.org
2 versions - Latest release: about 5 years ago - 1 dependent package - 178 dependent repositories - 21 thousand downloads last month - 9,315 stars on GitHub - 2 maintainers
torchgeometry 0.1.2 💰
differential geometric computer vision for deep learning2 versions - Latest release: about 5 years ago - 1 dependent package - 178 dependent repositories - 21 thousand downloads last month - 9,315 stars on GitHub - 2 maintainers
Top 1.6% on pypi.org
56 versions - Latest release: over 4 years ago - 2 dependent packages - 169 dependent repositories - 9.95 thousand downloads last month - 12,060 stars on GitHub - 1 maintainer
bert-serving-client 1.10.0
Mapping a variable-length sentence to a fixed-length vector using BERT model (Client)56 versions - Latest release: over 4 years ago - 2 dependent packages - 169 dependent repositories - 9.95 thousand downloads last month - 12,060 stars on GitHub - 1 maintainer
Related Keywords
deep-learning
1,581
machine-learning
1,139
python
888
tensorflow
557
deep learning
301
computer-vision
300
nlp
265
learning
220
neural-network
201
object-detection
200
transformer
194
data-science
190
ai
184
transformers
183
neural-networks
178
torch
173
machine learning
173
natural-language-processing
171
keras
161
ml
155
artificial-intelligence
152
image-classification
139
deep
139
reinforcement-learning
128
mlops
125
onnx
119
AI
117
jax
106
gpu
102
numpy
102
pytorch-lightning
101
machine
100
automl
98
bert
97
vision
94
yolov5
90
artificial intelligence
90
NLP
90
pretrained-models
89
scikit-learn
88
python3
87
deep-neural-networks
86
llm
85
cuda
82
language-model
78
classification
77
visualization
73
computer vision
72
hacktoberfest
71
hyperparameter-optimization
70
image-processing
68
instance-segmentation
66
deeplearning
65
dataset
64
yolov8
63
inference
62
optimization
62
mxnet
61
segmentation
61
spark
59
cnn
58
semantic-segmentation
56
time-series
54
PyTorch
54
transfer-learning
54
yolo
53
huggingface
53
graph-neural-networks
52
distributed
51
speech-recognition
49
multimodal
48
convolutional-neural-networks
48
neural-architecture-search
47
speech
46
datasets
46
framework
44
distributed-deep-learning
44
metrics
44
resnet
43
kubernetes
43
tensorboard
43
models
42
audio
42
forecasting
42
fastai
42
BERT
42
jupyter-notebook
42
tensor
42
automated-machine-learning
41
grounding-dino
41
labeling-tool
41
tabular-data
41
training
40
foundation-models
39
image-annotation
39
segment-anything
38
model-distillation
38
gan
38
auto-labeling
37
gpt
37