Machine Learning & Deep Learning Packages
Machine-learning and deep-learning frameworks and model libraries for training, fine-tuning, and serving models (PyTorch, scikit-learn, Transformers).
Packages in Machine Learning & Deep Learning
Kind
Sort:Used by Most Apps
Optimum
Hugging Face's optimization toolkit for accelerating training and inference of Transformers, Diffusers, TIMM, and Sentence-Transformers.
human
A TensorFlow.js library for real-time face, body, and hand detection, iris tracking, and age, gender, and emotion prediction in the browser or Node.js.
hdbscan
A high-performance, scikit-learn-compatible implementation of HDBSCAN for density-based clustering with automatic cluster count detection.
rust-bert
Rust-native NLP pipelines and transformer models, ported from Hugging Face's Transformers library
Thinc
A lightweight, type-checked functional deep learning library that wraps PyTorch, TensorFlow, and MXNet
BlingFire
A lightning-fast C++ finite-state-machine tokenizer for NLP text processing, exposed as a zero-configuration Python library.
BrowserGym Core
Gymnasium environment for building and evaluating LLM-driven web agents in a real Chromium browser.
tomesd
Speed up Stable Diffusion by merging redundant tokens with no retraining
TensorFlow Text
Text-processing ops -- tokenizers, Unicode normalization, and more -- that run natively inside the TensorFlow graph.
SacreBLEU
Hassle-free, reproducible BLEU, chrF, and TER scores for machine translation
clean-fid
Compute reliable FID and KID scores in PyTorch with correct image resizing and quantization
TensorFlow I/O
Extra file systems and file formats for TensorFlow, integrated directly into tf.data pipelines.
llama-cpp-2
Safe, up-to-date Rust bindings to llama.cpp for local LLM inference
S3Tokenizer
A pure PyTorch reimplementation of CosyVoice's supervised semantic speech tokenizer, built for fast batch inference and long-audio speech-code extraction.
misaki
A multilingual grapheme-to-phoneme engine built for the Kokoro text-to-speech models
sklearn-crfsuite
A scikit-learn compatible wrapper for CRFsuite that brings familiar fit/predict sequence labeling to Python.
nnnoiseless
Recurrent-neural-network audio noise suppression in Rust, a safe port of Xiph's RNNoise.
ml_dtypes
Stand-alone NumPy dtype extensions for machine learning: bfloat16, float8 variants, MX floats, and narrow ints.
rasa-sdk
Python SDK for building the custom action server that connects a Rasa chatbot to your own APIs and business logic.
Spandrel
Load and run pre-trained PyTorch super-resolution and restoration models with automatic architecture detection.
arroy
Annoy-inspired approximate nearest neighbors in Rust, backed by LMDB for low memory usage.
transcribe-rs
Multi-engine speech-to-text library for Rust with local and remote backends.
Comfy Kitchen
A fast PyTorch kernel library for diffusion inference with multiple compute backends.
cython-blis
Fast BLAS-style matrix multiplication for Python, bundled as a self-contained C extension with no system BLAS dependency.
