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
dspy
DSPy replaces brittle prompt engineering with compositional Python code, letting optimizers tune your language model pipeline's prompts and weights automatically.
GFPGAN
Restore realistic detail to low-quality faces in PyTorch using a pretrained GAN prior
Gym
Standard Python API and environments for reinforcement learning
jaxlib
The compiled XLA/PJRT runtime and native kernel library that powers JAX's array computation and program transformations.
InsightFace
A Python library for state-of-the-art 2D and 3D face detection, recognition, and alignment.
fastai
A high-level deep learning framework that layers state-of-the-art training APIs on top of PyTorch.
shap
Game-theoretic Shapley values to explain the output of any machine learning model
surya
A 650M-parameter vision-language OCR model that handles document layout analysis, reading order, and table recognition across 90+ languages.
prophet
Automatic forecasting library for time series with strong seasonality, using an interpretable additive model behind a scikit-learn-style API.
tfjs
A hardware-accelerated JavaScript library for training and deploying machine learning models in the browser and Node.js.
DeepEval
A Pytest-like open-source framework for unit-testing and evaluating LLM applications with research-backed metrics.
gensim
Topic modelling, document indexing, and similarity retrieval for Python.
stable-baselines3
Reliable PyTorch implementations of reinforcement learning algorithms with a consistent, sklearn-like API.
Presidio Analyzer
The PII detection engine behind Presidio, combining NER, regex, and checksum recognizers
PyOD
The most comprehensive Python library for anomaly and outlier detection, with 60+ detectors across tabular, time series, graph, text, image, and audio data.
TextBlob
Simple, Pythonic API for NLP: sentiment analysis, POS tagging, noun phrases, translation
BasicSR
An open-source PyTorch toolbox for image and video restoration, covering super-resolution, denoising, deblurring, and face restoration.
UMAP
A fast, scikit-learn-compatible dimension reduction library for visualizing and preprocessing high-dimensional data.
GEPA
Optimize prompts, code, and agents with LLM reflection and evolutionary search
ClearML
Python SDK for ML experiment tracking, dataset versioning, and MLOps pipeline orchestration.
torchdiffeq
GPU-accelerated, differentiable ODE solvers for PyTorch with O(1)-memory adjoint backpropagation for neural ODEs.
TensorFlow Datasets
Ready-to-use, versioned public datasets exposed as tf.data.Dataset objects, with first-class support for TensorFlow, JAX, and NumPy pipelines.
USearch
Smaller and faster single-file vector search and clustering engine
TensorFlow Hub
A Python library for downloading and reusing pretrained TensorFlow SavedModels for transfer learning with a minimum amount of code.
