Python machine learning libraries

443 libraries, 199 updated this year

Python holds well over a third of the list, and almost every major framework has its primary interface here. PyTorch, TensorFlow, JAX and Keras for deep learning; scikit-learn, XGBoost and LightGBM for classical models; Hugging Face Transformers for pretrained models; and NumPy, pandas and matplotlib underneath all of them.

The section is also the list's oldest and most mixed. Next to maintained projects sit research prototypes, single-paper implementations and notebooks from a decade ago. Sort by stars to see the established libraries, sort by recently updated to see what is active now, and turn on the archived filter to drop the projects their owners have frozen.

Use the task filter to narrow the page to computer vision, NLP, reinforcement learning or data visualization.

443 libraries

TensorFlowtensorflow/tensorflow

Open source software library for numerical computation using data flow graphs.

GeneralDeep learning
201.2k updated today

Transformershuggingface/transformers

A deep learning library containing thousands of pre-trained models on different tasks. The goto place for anything related to Large Language Models.

Deep learningLLMsNLPSpeech
168.1k updated today

PyTorchpytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

GeneralDeep learning
105k updated today

Supersetapache/incubator-superset

A data exploration platform designed to be visual, intuitive, and interactive.

Data & viz
75.3k updated today

scikit-learnscikit-learn/scikit-learn

A Python module for machine learning built on top of SciPy.

GeneralData & viz
67.9k updated today

keraskeras-team/keras

High-level neural networks frontend for TensorFlow, CNTK and Theano.

GeneralDeep learning
64.4k updated 1d ago

face_recognitionageitgey/face_recognition

Face recognition library that recognizes and manipulates faces from Python or from the command line.

Vision
56.9k updated 4mo ago

Pandaspandas-dev/pandas

A library providing high-performance, easy-to-use data structures and data analysis tools.

Data & viz
50.3k updated today

Streamlitstreamlit/streamlit

Streamlit is an framework to create beautiful data apps in hours, not weeks.

GeneralDeep learningData & viz
46.1k updated today

MLlib in Apache Sparkapache/spark

Distributed machine learning library in Spark.

GeneralData & viz
44.3k updated today

RLlibray-project/ray

RLlib is an industry level, highly scalable RL library for tf and torch, based on Ray. It's used by companies like Amazon and Microsoft to solve real-world decision making problems at scale.

Deep learningRL
44.1k updated today

Gradiogradio-app/gradio

A Python library for quickly creating and sharing demos of models. Debug models interactively in your browser, get feedback from collaborators, and generate public links without deploying anything.

GeneralData & viz
43.9k updated 1d ago

Colossal-AIhpcaitech/ColossalAI

An open-source deep learning system for large-scale model training and inference with high efficiency and low cost.

GeneralDeep learning
41.4k updated 2d ago

MindsDBmindsdb/mindsdb

Open Source framework to streamline use of neural networks.

GeneralDeep learning
39.8k updated 1d ago

timmrwightman/pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more.

Deep learningVision
37.2k updated 4d ago

JAXgoogle/jax

JAX is Autograd and XLA, brought together for high-performance machine learning research.

GeneralDeep learning
36.5k updated today

jiebafxsjy/jieba

Chinese Words Segmentation Utilities.

NLP
35.2k updated 2y ago

detectron2facebookresearch/detectron2

FAIR's next-generation research platform for object detection and segmentation. It is a ground-up rewrite of the previous version, Detectron, and is powered by the PyTorch deep learning framework.

Deep learningVision
34.9k updated 4d ago

CaffeBVLC/caffe

A deep learning framework developed with cleanliness, readability, and speed in mind.

GeneralDeep learning
34.6k updated 2y ago

OpenposeCMU-Perceptual-Computing-Lab/openpose

A real-time multi-person keypoint detection library for body, face, hands, and foot estimation

Vision
34.5k updated 2y ago

spaCyexplosion/spaCy

Industrial strength NLP with Python and Cython.

NLP
34k updated 1d ago

NumPynumpy/numpy

A fundamental package for scientific computing with Python.

Data & viz
33.1k updated today

ML-From-Scratcheriklindernoren/ML-From-Scratch

Implementations of Machine Learning models from scratch in Python with a focus on transparency. Aims to showcase the nuts and bolts of ML in an accessible way.

GeneralDeep learningRL
33.1k updated 3y ago

PyTorch LightningPyTorchLightning/pytorch-lightning

The lightweight PyTorch wrapper for high-performance AI research.

GeneralDeep learning
31.4k updated 12d ago

data-science-ipython-notebooksdonnemartin/data-science-ipython-notebooks

Continually updated Data Science Python Notebooks: Spark, Hadoop MapReduce, HDFS, AWS, Kaggle, scikit-learn, matplotlib, pandas, NumPy, SciPy, and various command lines.

Examples
29.4k updated 2y ago

MLXml-explore/mlx

MLX is an array framework for machine learning on Apple silicon, developed by Apple machine learning research.

Vision
28.9k updated 1d ago

XGBoostdmlc/xgboost

A parallelized optimized general purpose gradient boosting library.

General
28.9k updated 1d ago

Fastaifastai/fastai

High-level wrapper built on the top of Pytorch which supports vision, text, tabular data and collaborative filtering.

GeneralDeep learning
28.2k updated 1mo ago

Bayesian Methods for HackersCamDavidsonPilon/Probabilistic-Programming-and-Bayesian-Methods-for-Hackers

Book/iPython notebooks on Probabilistic Programming in Python.

General
28.2k updated 2y ago

Haystackdeepset-ai/haystack

A framework for building industrial-strength applications with Transformer models and LLMs.

LLMsNLP
26.8k updated today

Detectronfacebookresearch/Detectron Archived

FAIR's software system that implements state-of-the-art object detection algorithms, including Mask R-CNN. It is written in Python and powered by the Caffe2 deep learning framework.

Deep learningVision
26.3k archived

handsonmlageron/handson-ml

Fundamentals of machine learning in python.

Examples
25.6k updated 5mo ago

Pydata bookwesm/pydata-book

Materials and IPython notebooks for "Python for Data Analysis" by Wes McKinney, published by O'Reilly Media

Examples
25k updated 1y ago

Homemade Machine Learningtrekhleb/homemade-machine-learning

Python examples of popular machine learning algorithms with interactive Jupyter demos and math being explained

Examples
24.8k updated 11mo ago

Dashplotly/dash

A framework for creating analytical web applications built on top of Plotly.js, React, and Flask

Data & viz
24.5k updated 1d ago

PyTorch Geometricpyg-team/pytorch_geometric

> Graph Neural Network Library for PyTorch.

GeneralDeep learning
24.1k updated 2d ago

deepfaceserengil/deepface

A lightweight face recognition and facial attribute analysis (age, gender, emotion and race) framework for Python covering cutting-edge models such as VGG-Face, FaceNet, OpenFace, DeepFace, DeepID, Dlib and ArcFace.

Vision
23.6k updated today
m

matplotlibmatplotlib/matplotlib

A Python 2D plotting library.

Data & viz
23.4k updated 1d ago

Opikcomet-ml/opik

Open source engineering platform to debug, evaluate, and monitor your LLM applications, RAG systems, and agentic workflows with comprehensive tracing, automated evaluations, and production-ready dashboards. (Source Code)

GeneralLLMs
22.9k updated today

CometLLMcomet-ml/comet-llm

Track, log, visualize and evaluate your LLM prompts and prompt chains.

LLMsNLP
22.9k updated today

Microsoft RecommendersMicrosoft/Recommenders

Examples and best practices for building recommendation systems, provided as Jupyter notebooks. The repo contains some of the latest state of the art algorithms from Microsoft Research as well as from other companies and institutions.

GeneralDeep learning
22k updated 3d ago

Jina AIjina-ai/serve

An easier way to build neural search in the cloud. Compatible with Jupyter Notebooks.

Deep learning
21.9k updated 1y ago

RasaRasaHQ/rasa

A "machine learning framework to automate text-and voice-based conversations."

NLP
21.4k updated 3mo ago

MXNetapache/incubator-mxnet Archived

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Go, JavaScript and more.

GeneralDeep learning
20.8k archived

bokehbokeh/bokeh

Interactive Web Plotting for Python.

Data & viz
20.5k updated today

prophetfacebook/prophet

Fast and automated time series forecasting framework by Facebook.

General
20.5k updated 2d ago

ziplinequantopian/zipline

A Pythonic algorithmic trading library.

Data & viz
20.2k updated 2y ago

plotlyplotly/plotly.py

Collaborative web plotting for Python and matplotlib.

Data & viz
18.9k updated today

Neural Networks and Deep Learningmnielsen/neural-networks-and-deep-learning

Code samples for my book "Neural Networks and Deep Learning".

GeneralDeep learning
18k updated 2y ago

CNTKMicrosoft/CNTK Archived

The Computational Network Toolkit (CNTK) by Microsoft Research, is a unified deep-learning toolkit that describes neural networks as a series of computational steps via a directed graph.

GeneralDeep learning
17.6k archived
About this site
Esc

Type to search every entry. ↑↓ to move, Enter to open.