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
Whisper
A general-purpose speech recognition model for transcription, translation, and language identification, trained on large-scale weak supervision.
Gradio
Build and share machine learning web apps and demos in pure Python.
TRL
Fine-tune and align language models with SFT, DPO, GRPO, and RLHF trainers built on Hugging Face Transformers.
LanceDB
Serverless, embedded vector database for multimodal AI search
bitsandbytes
8-bit and 4-bit quantization primitives that make large language models fit and train on far less GPU memory.
ort
A safe Rust wrapper for ONNX Runtime — fast ML inference and training
Replicate
Official JavaScript/TypeScript client for running and managing machine learning models on Replicate's cloud API.
Docling
Parses PDF, DOCX, PPTX, HTML, images, audio, and more into a unified document representation built for generative AI pipelines.
diffusers
Hugging Face's modular Python toolbox of pretrained diffusion pipelines, models, and schedulers for image, video, and audio generation.
XGBoost
Scalable, portable gradient boosting library for Python, R, Java, Scala, and C++
faster-whisper
Fast, low-memory Whisper speech-to-text transcription powered by CTranslate2.
Magika
AI-powered file content type detection for Rust, identifying 200+ formats in milliseconds on a single CPU.
unstructured
Open-source Python library for partitioning and preprocessing unstructured documents for LLM pipelines
wandb
Official Python SDK and CLI for the Weights & Biases ML experiment tracking platform
RapidOCR
A fast, offline Python OCR library with pluggable ONNX Runtime, OpenVINO, and PaddlePaddle inference backends.
PySpark
Python API for Apache Spark, the unified engine for large-scale distributed data processing
jieba
Chinese text segmentation library for Python with precise, full, and search-engine cut modes
Transformers.js
Run Hugging Face Transformers models directly in the browser with no server or Python runtime required.
TensorBoard
TensorFlow's visualization toolkit for inspecting and debugging machine-learning runs
Evaluate
Hugging Face's library for evaluating machine learning models and datasets
gguf
The Python library for reading and writing GGUF, the single-file binary model format behind llama.cpp and the wider local-LLM ecosystem.
timm (PyTorch Image Models)
The largest collection of pretrained PyTorch image models, layers, and training utilities.
PyTorch Lightning
A deep learning framework that organizes PyTorch training code and automates multi-GPU, mixed-precision, and distributed training.
MLflow
Open source AI engineering platform for tracking, evaluating, and shipping ML models, LLMs, and agents to production.
