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

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
SWE-bench
swesmith

[NeurIPS 2025 D&B Spotlight] Scaling Data for SWE-agents

14M 639 117
pytorch
torchao

PyTorch native quantization and sparsity for training and inference

3.4M 3K 502
huggingface
optimum

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

1.7M 3K 639
aws
sagemaker-training

Train machine learning models within a 🐳 Docker container using 🧠 Amazon SageMaker.

233K 535 140
pytorch
torchft-nightly

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

201K 500 62
SwanHubX
swanlab

⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.

183K 4K 203
dstackai
dstack

Vendor-agnostic orchestration for training, inference and agentic workloads across NVIDIA, AMD, TPU, and Tenstorrent on clouds, Kubernetes, and bare metal.

174K 2K 224
jfilter
split-folders

🗂 Split folders with files (i.e. images) into training, validation and test (dataset) folders

32K 422 67
run-house
kubetorch

Distribute and run AI workloads on Kubernetes magically in Python, like PyTorch for ML infra.

29K 1K 57
run-house
runhouse

Distribute and run AI workloads on Kubernetes magically in Python, like PyTorch for ML infra.

26K 1K 57
ModelCloud
tokenicer

A (nicer) tokenizer you want to use for model inference and training: with all known peventable gotchas normalized or auto-fixed.

26K 11 4
hearmeneigh
datasetrising

Toolchain for creating custom datasets and training Stable Diffusion (1.x, 2.x, XL) models and LoRAs

9K 18 1
Demonstrandum
tensorbored-nightly

TensorBored: Fixing TensorBoard's deficiencies

6K 1 0
Chris-hughes10
pytorch-accelerated

A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop which is flexible enough to handle the majority of use cases, and capable of utilizing different hardware options with no code changes required. Docs: https://pytorch-accelerated.readthedocs.io/en/latest/

5K 192 19
ContinualAI
avalanche-lib

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.

4K 2K 316
hitlic
deepepochs

Pytorch模型训练工具

4K 39 1
bytedance
lightseq

LightSeq: A High Performance Library for Sequence Processing and Generation

4K 3K 333
inclusionAI
awex

A high-performance RL training-inference weight synchronization framework, designed to enable second-level parameter updates from training to inference in RL workflows

3K 150 17
neptune-ai
neptune-sklearn

Experiment tracking for scikit-learn. 🧩 Log, organize, visualize and compare model metrics, parameters, dataset versions, and more.

3K 7 3
Goekdeniz-Guelmez
mlx-lm-lora

Train LLMs on Apple silicon with MLX and the Hugging Face Hub

3K 335 42
szymonmaszke
torchtraining-nightly

Functional & flexible neural network training with PyTorch.

2K 4 1
nuhame
mlpug

MLPug is a library for training and evaluating Machine Learning (ML) models, able to use different ML libraries as backends

2K 3 2
neptune-ai
neptune-tensorflow-keras

💡 Experiment tracking for TensorFlow/Keras. Log, organize, and compare model metrics, learning curves, hyperparameters, dataset versions, and more.

2K 6 5
M4cs
easymodels

Easily find and view pre-trained AI models and deep learning projects through the command line :computer:

2K 191 11
    • Data from PyPI, GitHub, ClickHouse, and BigQuery