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
PyTorch
A Python-first tensor library with GPU acceleration and a dynamic, define-by-run autograd engine for building and training deep neural networks.
Transformers
State-of-the-art pretrained models for text, vision, audio, and multimodal inference and training.
scikit-learn
Simple and efficient tools for machine learning and data analysis in Python.
tiktoken
OpenAI's fast, Rust-backed BPE tokenizer for counting and encoding text into the exact tokens GPT models see.
huggingface_hub
The official Python client and hf CLI for the Hugging Face Hub, for downloading, uploading, and managing models, datasets, and Spaces.
Datasets
One-line loading and fast, Arrow-backed processing for thousands of ML datasets
sentence-transformers
A Python library for computing state-of-the-art sentence, text, and image embeddings, rerankers, and sparse and multi-vector representations.
ONNX Runtime
Cross-platform, high-performance ML inference and training accelerator for ONNX models
Tokenizers
Fast, production-grade tokenizer implementations for NLP and LLMs, written in Rust
accelerate
A thin PyTorch wrapper that runs the same raw training script on CPU, multi-GPU, or TPU with mixed precision, DeepSpeed, and FSDP support.
vision
PyTorch's official library for datasets, pretrained models, and image transformations in computer vision.
sentencepiece
A fast, language-independent subword tokenizer and detokenizer trained directly from raw sentences, purpose-built for neural network text generation and LLM pipelines.
qdrant-client
Official Python SDK for the Qdrant vector search engine
opencv-python
Pre-built, pip-installable OpenCV bindings that give Python programs full computer vision and image-processing capabilities without compiling from source.
safetensors
A safe, zero-copy binary format for storing and loading ML tensors without pickle
PEFT
Hugging Face's library of state-of-the-art parameter-efficient fine-tuning methods for large models.
torchaudio
GPU-accelerated audio processing library built on PyTorch, with differentiable transforms, Kaldi-compatible features, and pretrained speech models.
pymilvus
The official Python SDK for Milvus, the open-source vector database for embedding-based similarity search.
spaCy
Industrial-strength Natural Language Processing library for Python
google-cloud-aiplatform
Google's official Python SDK for Vertex AI model training, deployment, and generative AI workflows
tesseract.js
Pure JavaScript OCR library that recognizes text in over 100 languages, running in the browser or Node.js via WebAssembly.
onnxruntime-node
Microsoft's official Node.js binding for running ONNX model inference in production JavaScript apps
Transformers.js
Run Hugging Face Transformers models directly in the browser or Node.js with no server required.
einops
A readable, framework-independent notation for tensor rearrangement, reduction, and repetition.
