General-purpose ML libraries

609 libraries, 201 updated this year

The broadest part of the list: libraries that train and run models without being tied to one kind of data. It mixes two generations. Classical toolkits such as scikit-learn, XGBoost, LightGBM and CatBoost fit models to tables of features and remain the first choice for structured business data. Deep learning frameworks such as TensorFlow, PyTorch, JAX and Keras build neural networks of any shape and run them on GPUs.

Plenty of entries here are wrappers, bindings or research code from the 2010s that no longer receive commits. The activity column separates the projects that still ship releases from those that stopped, and the archived filter hides repositories their owners have frozen. A library listed under several languages, like XGBoost or MXNet, appears once with every language it serves.

For a new project in Python, scikit-learn plus one gradient boosting library covers most tabular problems, and PyTorch is the common default for neural networks. Use the task filter to narrow this page to deep learning, or open a language page to see what exists outside Python.

609 libraries

TensorFlowtensorflow/tensorflow

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

PythonDeep learning
201.2k updated today

PyTorchpytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PythonDeep learning
105k updated today

OpenCVopencv/opencv

OpenCV has C++, C, Python, Java and MATLAB interfaces and supports Windows, Linux, Android and Mac OS.

C++Deep learningVision
91.3k updated today

scikit-learnscikit-learn/scikit-learn

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

PythonData & viz
67.9k updated today

keraskeras-team/keras

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

PythonDeep learning
64.4k updated 1d ago

Streamlitstreamlit/streamlit

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

PythonDeep learningData & viz
46.1k updated today

MLlib in Apache Sparkapache/spark

Distributed machine learning library in Spark.

Data & viz
44.3k 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.

PythonData & 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.

PythonDeep learning
41.4k updated 2d ago

MindsDBmindsdb/mindsdb

Open Source framework to streamline use of neural networks.

PythonDeep learning
39.8k updated 1d ago

JAXgoogle/jax

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

PythonDeep learning
36.5k updated today

CaffeBVLC/caffe

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

Deep learning
34.6k updated 2y ago

Netronlutzroeder/netron

Visualizer for machine learning models.

JavaScriptDeep learningMLOps
33.6k 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.

PythonDeep learningRL
33.1k updated 3y ago

PyTorch LightningPyTorchLightning/pytorch-lightning

The lightweight PyTorch wrapper for high-performance AI research.

PythonDeep learning
31.4k updated 12d ago

XGBoostdmlc/xgboost

A parallelized optimized general purpose gradient boosting library.

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.

PythonDeep 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.

Python
28.2k updated 2y ago

Darknetpjreddie/darknet

Darknet is an open source neural network framework written in C and CUDA. It is fast, easy to install, and supports CPU and GPU computation.

CDeep learning
26.5k updated 2y ago

PyTorch Geometricpyg-team/pytorch_geometric

> Graph Neural Network Library for PyTorch.

PythonDeep learning
24.1k updated 2d 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)

LLMs
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.

PythonDeep learning
22k updated 3d ago

candlehuggingface/candle

Candle is a minimalist ML framework for Rust with a focus on performance (including GPU support) and ease of use.

Rust
21.2k updated 10d 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.

+3Deep learning
20.8k archived

prophetfacebook/prophet

Fast and automated time series forecasting framework by Facebook.

Python
20.5k updated 2d ago

TensorFlow.jstensorflow/tfjs

A WebGL accelerated, browser based JavaScript library for training and deploying ML models.

JavaScriptDeep learning
19.2k updated 4mo ago

LightGBMMicrosoft/LightGBM

Microsoft's fast, distributed, high performance gradient boosting (GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.

C++
18.9k updated today

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

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

PythonDeep 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.

Deep learning
17.6k archived

Awesome TensorFlowjtoy/awesome-tensorflow

A list of all things related to TensorFlow.

Learning
17.5k updated 8mo ago

QuestDBquestdb/questdb

A relational column-oriented database designed for real-time analytics on time series and event data.

C++
17.5k updated today

gensimRaRe-Technologies/gensim

Topic Modelling for Humans.

PythonNLP
16.5k updated 11mo ago

numpy-MLddbourgin/numpy-ml

Reference implementations of ML models written in numpy

PythonDeep learningNLP
16.3k updated 2y ago

Optunaoptuna/optuna

Optuna is an automatic hyperparameter optimization software framework, particularly designed for machine learning.

PythonMLOps
15k updated 1d ago

Brain.jsBrainJS/brain.js

Neural networks in JavaScript - continued community fork of Brain.

JavaScriptDeep learning
14.9k updated 2y ago

DLibdavisking/dlib

DLib has C++ and Python interfaces for face detection and training general object detectors.

C++Deep learningVisionData & viz
14.5k updated 3d ago

Annoyspotify/annoy

Approximate nearest neighbours implementation.

Python
14.3k updated 12mo ago

PhoenixArize-ai/phoenix

Uncover insights, surface problems, monitor and fine tune your generative LLM, CV and tabular models.

C++LLMs
12k updated today

cleanlabcleanlab/cleanlab

The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

Python
11.7k updated 9mo ago

Convnet.jskarpathy/convnetjs Deprecated

ConvNetJS is a JavaScript library for training Deep Learning models

JavaScriptDeep learning
11.2k updated 3y ago

Turi Createapple/turicreate Archived

Machine learning from Apple. Turi Create simplifies the development of custom machine learning models. You don't have to be a machine learning expert to add recommendations, object detection, image classification, image similarity or activity classification to your app.

PythonDeep learningVision
11.2k archived

AutoGluonawslabs/autogluon

AutoML for Image, Text, Tabular, Time-Series, and MultiModal Data.

PythonDeep learningVisionNLPMLOps
10.8k updated 2d ago

sktimealan-turing-institute/sktime

A unified framework for machine learning with time series

Python
10.1k updated today

PyODyzhao062/pyod

> Python Outlier Detection, comprehensive and scalable Python toolkit for detecting outlying objects in multivariate data. Featured for Advanced models, including Neural Networks/Deep Learning and Outlier Ensembles.

PythonDeep learningLLMs
10.1k updated 12d ago

TPOTEpistasisLab/tpot

Tool that automatically creates and optimizes machine learning pipelines using genetic programming. Consider it your personal data science assistant, automating a tedious part of machine learning.

PythonMLOps
10.1k updated 1y ago

PySyftOpenMined/PySyft

A Python library for secure and private Deep Learning built on PyTorch and TensorFlow.

PythonDeep learning
10k updated today

TheanoTheano/Theano

Optimizing GPU-meta-programming code generating array oriented optimizing math compiler in Python.

Python
10k updated 2y ago

PyCaretpycaret/pycaret

An open-source, low-code machine learning library in Python that automates machine learning workflows.

C++MLOps
9.9k updated 3mo ago

einopsarogozhnikov/einops

Deep learning operations reinvented (for pytorch, tensorflow, jax and others).

PythonDeep learning
9.6k updated 1mo ago

TFLearntflearn/tflearn

Deep learning library featuring a higher-level API for TensorFlow.

PythonDeep learning
9.6k updated 2y ago
Esc

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