14,357 dependents
| Package | Description | Downloads/month |
|---|---|---|
| SGLang is a high-performance serving framework for large language models and mul... | 287.7M | |
| Datasets, Transforms and Models specific to Computer Vision | 34.9M | |
| State-of-the-Art Text Embeddings | 25.7M | |
| 🚀 A simple way to launch, train, and use PyTorch models on almost any device and... | 24.8M | |
| The largest collection of PyTorch image encoders / backbones. Including train, e... | 13.2M | |
| Machine learning metrics for distributed, scalable PyTorch applications. | 12.7M | |
| Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code ... | 11.5M | |
| 🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning. | 10.8M | |
| Ultralytics YOLO 🚀 | 10.6M | |
| A high-throughput and memory-efficient inference and serving engine for LLMs | 9.4M | |
| Accessible large language models via k-bit quantization for PyTorch. | 6.5M | |
| Fast, Flexible and Portable Structured Generation | 6.4M | |
| A safetensors extension to efficiently store sparse quantized tensors on disk | 6.3M | |
| Profile PyTorch models for FLOPs and parameters, helping to evaluate computation... | 5.9M | |
| FlashInfer: Kernel Library for LLM Serving | 4M | |
| XFormers: A collection of composable Transformer building blocks. | 3.5M | |
| torch c dlpack ext | 3.4M | |
| 🐍 Geometric Computer Vision Library for Spatial AI | 3.3M | |
| An open source implementation of CLIP. | 3.1M | |
| 3M | ||
| MTEB: Massive Text Embedding Benchmark | 2.7M | |
| Ready-to-use OCR with 80+ supported languages and all popular writing scripts in... | 2.7M | |
| TensorDict is a pytorch dedicated tensor container. | 2.5M | |
| This package contains the AI models used by the Docling PDF conversion package | 2.4M | |
| Robust Speech Recognition via Large-Scale Weak Supervision | 2.3M | |
| A small package to create visualizations of PyTorch execution graphs | 2.3M | |
| A PyTorch repo for data loading and utilities to be shared by the PyTorch domain... | 2.2M | |
| State-of-the-art speaker diarization toolkit | 2.2M | |
| Web UI for training and running open models like Gemma 4, Qwen3.6, DeepSeek, gpt... | 1.9M | |
| Give your project support for a variety of PyTorch model architectures, includin... | 1.8M | |
| Fast audio data augmentation in PyTorch. Inspired by audiomentations. Useful for... | 1.7M | |
| Pitch-shift audio clips quickly with PyTorch (CUDA supported)! Additional utilit... | 1.7M | |
| Vector (and Scalar) Quantization, in Pytorch | 1.7M | |
| 🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sente... | 1.7M | |
| All-in-one speech toolkit in pure Python and Pytorch | 1.6M | |
| SageMaker Serve package for model serving and deployment | 1.6M | |
| Web UI for training and running open models like Gemma 4, Qwen3.6, DeepSeek, gpt... | 1.4M | |
| Fast and memory-efficient exact attention | 1.4M | |
| A concise but complete full-attention transformer with a set of promising experi... | 1.4M | |
| A modular, primitive-first, python-first PyTorch library for Reinforcement Learn... | 1.3M | |
| DeepSpeed is a deep learning optimization library that makes distributed trainin... | 1.3M | |
| PyTorch version of Stable Baselines, reliable implementations of reinforcement l... | 1.2M | |
| Deep universal probabilistic programming with Python and PyTorch | 1.2M | |
| Count the MACs / FLOPs of your PyTorch model. | 1.1M | |
| A library for performing inference using trained models. | 1.1M | |
| WhisperX: Automatic Speech Recognition with Word-level Timestamps (& Diarizatio... | 1.1M | |
| A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch | 1M | |
| A PyTorch-native inference engine with cache, parallelism, quantization for Diff... | 1M | |
| The fastai deep learning library | 994K | |
| Stanford NLP Python library for tokenization, sentence segmentation, NER, and pa... | 942K |
