Top 50 Awesome List

josephmisiti/awesome-machine-learning

Computer Science  5 days ago  56.1k
A curated list of awesome Machine Learning frameworks, libraries and software.
View byDAY/WEEK/README
View on Github

Awesome Machine Learning Awesome Track Awesome List

A curated list of awesome machine learning frameworks, libraries and software (by language). Inspired by awesome-php.

If you want to contribute to this list (please do), send me a pull request or contact me @josephmisiti. Also, a listed repository should be deprecated if:

  • Repository's owner explicitly says that "this library is not maintained".
  • Not committed for a long time (2~3 years).

Further resources:

  • For a list of free machine learning books available for download, go here.

  • For a list of professional machine learning events, go here.

  • For a list of (mostly) free machine learning courses available online, go here.

  • For a list of blogs and newsletters on data science and machine learning, go here.

  • For a list of free-to-attend meetups and local events, go here.

Table of Contents

Frameworks and Libraries

Tools

Credits

APL

General-Purpose Machine Learning

  • naive-aplstars21 - Naive Bayesian Classifier implementation in APL. [Deprecated]

C

General-Purpose Machine Learning

  • Darknetstars23.3k - 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.
  • Recommenderstars252 - A C library for product recommendations/suggestions using collaborative filtering (CF).
  • Hybrid Recommender Systemstars11 - A hybrid recommender system based upon scikit-learn algorithms. [Deprecated]
  • neonrvmstars33 - neonrvm is an open source machine learning library based on RVM technique. It's written in C programming language and comes with Python programming language bindings.
  • cONNXrstars134 - An ONNX runtime written in pure C (99) with zero dependencies focused on small embedded devices. Run inference on your machine learning models no matter which framework you train it with. Easy to install and compiles everywhere, even in very old devices.
  • libonnxstars421 - A lightweight, portable pure C99 onnx inference engine for embedded devices with hardware acceleration support.

Computer Vision

  • CCVstars6.9k - C-based/Cached/Core Computer Vision Library, A Modern Computer Vision Library.
  • VLFeat - VLFeat is an open and portable library of computer vision algorithms, which has a Matlab toolbox.

C++

Computer Vision

  • DLib - DLib has C++ and Python interfaces for face detection and training general object detectors.
  • EBLearn - Eblearn is an object-oriented C++ library that implements various machine learning models [Deprecated]
  • OpenCV - OpenCV has C++, C, Python, Java and MATLAB interfaces and supports Windows, Linux, Android and Mac OS.
  • VIGRAstars374 - VIGRA is a genertic cross-platform C++ computer vision and machine learning library for volumes of arbitrary dimensionality with Python bindings.
  • Openposestars25k - A real-time multi-person keypoint detection library for body, face, hands, and foot estimation

General-Purpose Machine Learning

  • Nebulgymstars175 - Easy-to-use library to accelerate AI training. [DEEP LEARNING]
  • Nebullvmstars1.4k - Easy-to-use library to boost AI inference. [DEEP LEARNING]
  • BanditLibstars130 - A simple Multi-armed Bandit library. [Deprecated]
  • Caffestars32.9k - A deep learning framework developed with cleanliness, readability, and speed in mind. [DEEP LEARNING]
  • CatBooststars6.7k - General purpose gradient boosting on decision trees library with categorical features support out of the box. It is easy to install, contains fast inference implementation and supports CPU and GPU (even multi-GPU) computation.
  • CNTKstars17.2k - 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.
  • CUDA - This is a fast C++/CUDA implementation of convolutional [DEEP LEARNING]
  • DeepDetectstars2.4k - A machine learning API and server written in C++11. It makes state of the art machine learning easy to work with and integrate into existing applications.
  • Distributed Machine learning Tool Kit (DMTK) - A distributed machine learning (parameter server) framework by Microsoft. Enables training models on large data sets across multiple machines. Current tools bundled with it include: LightLDA and Distributed (Multisense) Word Embedding.
  • DLib - A suite of ML tools designed to be easy to imbed in other applications.
  • DSSTNEstars4.4k - A software library created by Amazon for training and deploying deep neural networks using GPUs which emphasizes speed and scale over experimental flexibility.
  • DyNetstars3.3k - A dynamic neural network library working well with networks that have dynamic structures that change for every training instance. Written in C++ with bindings in Python.
  • Fidostars412 - A highly-modular C++ machine learning library for embedded electronics and robotics.
  • igraph - General purpose graph library.
  • Intel® oneAPI Data Analytics Librarystars521 - A high performance software library developed by Intel and optimized for Intel's architectures. Library provides algorithmic building blocks for all stages of data analytics and allows to process data in batch, online and distributed modes.
  • LightGBMstars14.2k - 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.
  • libfmstars1.4k - A generic approach that allows to mimic most factorization models by feature engineering.
  • MLDB - The Machine Learning Database is a database designed for machine learning. Send it commands over a RESTful API to store data, explore it using SQL, then train machine learning models and expose them as APIs.
  • mlpack - A scalable C++ machine learning library.
  • MXNetstars20.1k - Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Go, Javascript and more.
  • N2D2stars117 - CEA-List's CAD framework for designing and simulating Deep Neural Network, and building full DNN-based applications on embedded platforms
  • oneDNNstars3k - An open-source cross-platform performance library for deep learning applications.
  • ParaMontestars173 - A general-purpose library with C/C++ interface for Bayesian data analysis and visualization via serial/parallel Monte Carlo and MCMC simulations. Documentation can be found here.
  • proNet-corestars3 - A general-purpose network embedding framework: pair-wise representations optimization Network Edit.
  • PyCaretstars6.3k - An open-source, low-code machine learning library in Python that automates machine learning workflows.
  • PyCUDA - Python interface to CUDA
  • ROOT - A modular scientific software framework. It provides all the functionalities needed to deal with big data processing, statistical analysis, visualization and storage.
  • shark - A fast, modular, feature-rich open-source C++ machine learning library.
  • Shogunstars2.9k - The Shogun Machine Learning Toolbox.
  • sofia-ml - Suite of fast incremental algorithms.
  • Stan - A probabilistic programming language implementing full Bayesian statistical inference with Hamiltonian Monte Carlo sampling.
  • Timbl - A software package/C++ library implementing several memory-based learning algorithms, among which IB1-IG, an implementation of k-nearest neighbor classification, and IGTree, a decision-tree approximation of IB1-IG. Commonly used for NLP.
  • Vowpal Wabbit (VW)stars8k - A fast out-of-core learning system.
  • Warp-CTCstars4k - A fast parallel implementation of Connectionist Temporal Classification (CTC), on both CPU and GPU.
  • XGBooststars23.2k - A parallelized optimized general purpose gradient boosting library.
  • ThunderGBMstars639 - A fast library for GBDTs and Random Forests on GPUs.
  • ThunderSVMstars1.4k - A fast SVM library on GPUs and CPUs.
  • LKYDeepNNstars43 - A header-only C++11 Neural Network library. Low dependency, native traditional chinese document.
  • xLearnstars3k - A high performance, easy-to-use, and scalable machine learning package, which can be used to solve large-scale machine learning problems. xLearn is especially useful for solving machine learning problems on large-scale sparse data, which is very common in Internet services such as online advertising and recommender systems.
  • Featuretoolsstars6.3k - A library for automated feature engineering. It excels at transforming transactional and relational datasets into feature matrices for machine learning using reusable feature engineering "primitives".
  • skynetstars63 - A library for learning neural networks, has C-interface, net set in JSON. Written in C++ with bindings in Python, C++ and C#.
  • Feaststars3.6k - A feature store for the management, discovery, and access of machine learning features. Feast provides a consistent view of feature data for both model training and model serving.
  • Hopsworksstars793 - A data-intensive platform for AI with the industry's first open-source feature store. The Hopsworks Feature Store provides both a feature warehouse for training and batch based on Apache Hive and a feature serving database, based on MySQL Cluster, for online applications.
  • Polyaxonstars3.2k - A platform for reproducible and scalable machine learning and deep learning.
  • QuestDB A relational column-oriented database designed for real-time analytics on time series and event data.

Natural Language Processing

  • BLLIP Parserstars214 - BLLIP Natural Language Parser (also known as the Charniak-Johnson parser).
  • colibri-corestars121 - C++ library, command line tools, and Python binding for extracting and working with basic linguistic constructions such as n-grams and skipgrams in a quick and memory-efficient way.
  • CRF++ - Open source implementation of Conditional Random Fields (CRFs) for segmenting/labeling sequential data & other Natural Language Processing tasks. [Deprecated]
  • CRFsuite - CRFsuite is an implementation of Conditional Random Fields (CRFs) for labeling sequential data. [Deprecated]
  • frogstars70 - Memory-based NLP suite developed for Dutch: PoS tagger, lemmatiser, dependency parser, NER, shallow parser, morphological analyzer.
  • libfoliastars14 - C++ library for the FoLiA format
  • MeTAstars649 - MeTA : ModErn Text Analysis is a C++ Data Sciences Toolkit that facilitates mining big text data.
  • MIT Information Extraction Toolkitstars2.8k - C, C++, and Python tools for named entity recognition and relation extraction
  • uctostars58 - Unicode-aware regular-expression based tokenizer for various languages. Tool and C++ library. Supports FoLiA format.

Speech Recognition

  • Kaldistars12k - Kaldi is a toolkit for speech recognition written in C++ and licensed under the Apache License v2.0. Kaldi is intended for use by speech recognition researchers.

Sequence Analysis

  • ToPSstars34 - This is an object-oriented framework that facilitates the integration of probabilistic models for sequences over a user defined alphabet. [Deprecated]

Gesture Detection

  • grtstars786 - The Gesture Recognition Toolkit (GRT) is a cross-platform, open-source, C++ machine learning library designed for real-time gesture recognition.

Common Lisp

General-Purpose Machine Learning

Clojure

Natural Language Processing

General-Purpose Machine Learning

  • scicloj.mlstars127 - A idiomatic Clojure machine learning library based on tech.ml.dataset with a unique approach for immutable data processing pipelines.
  • clj-mlstars135 - A machine learning library for Clojure built on top of Weka and friends.
  • clj-boost - Wrapper for XGBoost
  • Touchstonestars133 - Clojure A/B testing library.
  • Clojushstars321 - The Push programming language and the PushGP genetic programming system implemented in Clojure.
  • lambda-mlstars71 - Simple, concise implementations of machine learning techniques and utilities in Clojure.
  • Inferstars176 - Inference and machine learning in Clojure. [Deprecated]
  • Encogstars136 - Clojure wrapper for Encog (v3) (Machine-Learning framework that specializes in neural-nets). [Deprecated]
  • Fungpstars100 - A genetic programming library for Clojure. [Deprecated]
  • Statistikerstars63 - Basic Machine Learning algorithms in Clojure. [Deprecated]
  • clortexstars180 - General Machine Learning library using Numenta’s Cortical Learning Algorithm. [Deprecated]
  • comportexstars152 - Functionally composable Machine Learning library using Numenta’s Cortical Learning Algorithm. [Deprecated]

Deep Learning

  • MXNet - Bindings to Apache MXNet - part of the MXNet project
  • Deep Diamondstars377 - A fast Clojure Tensor & Deep Learning library
  • jutsu.aistars98 - Clojure wrapper for deeplearning4j with some added syntactic sugar.
  • cortexstars1.3k - Neural networks, regression and feature learning in Clojure.
  • Flarestars282 - Dynamic Tensor Graph library in Clojure (think PyTorch, DynNet, etc.)
  • dl4cljstars100 - Clojure wrapper for Deeplearning4j.

Data Analysis

  • tech.ml.datasetstars467 - Clojure dataframe library and pipeline for data processing and machine learning
  • Tableclothstars181 - A dataframe grammar wrapping tech.ml.dataset, inspired by several R libraries
  • Pantherastars180 - Clojure API wrapping Python's Pandas library
  • Incanter - Incanter is a Clojure-based, R-like platform for statistical computing and graphics.
  • PigPenstars541 - Map-Reduce for Clojure.
  • Genistars240 - a Clojure dataframe library that runs on Apache Spark

Data Visualization

  • Hanamistars303 : Clojure(Script) library and framework for creating interactive visualization applications based in Vega-Lite (VGL) and/or Vega (VG) specifications. Automatic framing and layouts along with a powerful templating system for abstracting visualization specs
  • Saitestars114 - Clojure(Script) client/server application for dynamic interactive explorations and the creation of live shareable documents capturing them using Vega/Vega-Lite, CodeMirror, markdown, and LaTeX
  • Ozstars783 - Data visualisation using Vega/Vega-Lite and Hiccup, and a live-reload platform for literate-programming
  • Envisionstars78 - Clojure Data Visualisation library, based on Statistiker and D3.
  • Pink Gorilla Notebookstars100 - A Clojure/Clojurescript notebook application/-library based on Gorilla-REPL
  • clojupyterstars748 - A Jupyter kernel for Clojure - run Clojure code in Jupyter Lab, Notebook and Console.
  • notespacestars139 - Notebook experience in your Clojure namespace
  • Delightstars268 - A listener that streams your spark events logs to delight, a free and improved spark UI

Interop

Misc

  • Neanderthal - Fast Clojure Matrix Library (native CPU, GPU, OpenCL, CUDA)
  • kixistatsstars327 - A library of statistical distribution sampling and transducing functions
  • fastmathstars168 - A collection of functions for mathematical and statistical computing, macine learning, etc., wrapping several JVM libraries
  • matlibstars23 - a Clojure library of optimisation and control theory tools and convenience functions based on Neanderthal.

Extra

  • Scicloj - Curated list of ML related resources for Clojure.

Crystal

General-Purpose Machine Learning

Elixir

General-Purpose Machine Learning

Natural Language Processing

  • Stemmerstars143 - An English (Porter2) stemming implementation in Elixir.

Erlang

General-Purpose Machine Learning

Fortran

General-Purpose Machine Learning

Read the paper here.

Data Analysis / Data Visualization

  • ParaMontestars173 - A general-purpose Fortran library for Bayesian data analysis and visualization via serial/parallel Monte Carlo and MCMC simulations. Documentation can be found here.

Go

Natural Language Processing

General-Purpose Machine Learning

  • birdlandstars29 - A recommendation library in Go.
  • eaoptstars812 - An evolutionary optimization library.
  • leavesstars354 - A pure Go implementation of the prediction part of GBRTs, including XGBoost and LightGBM.
  • gobrainstars526 - Neural Networks written in Go.
  • go-featureprocessingstars82 - Fast and convenient feature processing for low latency machine learning in Go.
  • go-mxnet-predictorstars55 - Go binding for MXNet c_predict_api to do inference with a pre-trained model.
  • go-ml-benchmarksstars24 — benchmarks of machine learning inference for Go.
  • go-ml-transpilerstars65 - An open source Go transpiler for machine learning models.
  • golearnstars8.5k - Machine learning for Go.
  • gomlstars1.4k - Machine learning library written in pure Go.
  • gorgoniastars4.7k - Deep learning in Go.
  • gorostars351 - A high-level machine learning library in the vein of Keras.
  • gorsestars6.2k - An offline recommender system backend based on collaborative filtering written in Go.
  • therfoostars18 - An embedded deep learning library for Go.
  • neatstars66 - Plug-and-play, parallel Go framework for NeuroEvolution of Augmenting Topologies (NEAT). [Deprecated]
  • go-prstars62 - Pattern recognition package in Go lang. [Deprecated]
  • go-mlstars195 - Linear / Logistic regression, Neural Networks, Collaborative Filtering and Gaussian Multivariate Distribution. [Deprecated]
  • GoNNstars348 - GoNN is an implementation of Neural Network in Go Language, which includes BPNN, RBF, PCN. [Deprecated]
  • bayesianstars742 - Naive Bayesian Classification for Golang. [Deprecated]
  • go-galibstars193 - Genetic Algorithms library written in Go / Golang. [Deprecated]
  • Cloudforeststars717 - Ensembles of decision trees in Go/Golang. [Deprecated]
  • go-dnnstars6 - Deep Neural Networks for Golang (powered by MXNet)

Spatial analysis and geometry

Data Analysis / Data Visualization

Computer vision

  • GoCVstars5k - Package for computer vision using OpenCV 4 and beyond.

Reinforcement learning

Haskell

General-Purpose Machine Learning

Java

Natural Language Processing

  • Cortical.io - Retina: an API performing complex NLP operations (disambiguation, classification, streaming text filtering, etc...) as quickly and intuitively as the brain.
  • IRIS - Cortical.io's FREE NLP, Retina API Analysis Tool (written in JavaFX!) - See the Tutorial Video.
  • CoreNLP - Stanford CoreNLP provides a set of natural language analysis tools which can take raw English language text input and give the base forms of words.
  • Stanford Parser - A natural language parser is a program that works out the grammatical structure of sentences.
  • Stanford POS Tagger - A Part-Of-Speech Tagger (POS Tagger).
  • Stanford Name Entity Recognizer - Stanford NER is a Java implementation of a Named Entity Recognizer.
  • Stanford Word Segmenter - Tokenization of raw text is a standard pre-processing step for many NLP tasks.
  • Tregex, Tsurgeon and Semgrex - Tregex is a utility for matching patterns in trees, based on tree relationships and regular expression matches on nodes (the name is short for "tree regular expressions").
  • Stanford Phrasal: A Phrase-Based Translation System
  • Stanford English Tokenizer - Stanford Phrasal is a state-of-the-art statistical phrase-based machine translation system, written in Java.
  • Stanford Tokens Regex - A tokenizer divides text into a sequence of tokens, which roughly correspond to "words".
  • Stanford Temporal Tagger - SUTime is a library for recognizing and normalizing time expressions.
  • Stanford SPIED - Learning entities from unlabeled text starting with seed sets using patterns in an iterative fashion.
  • Twitter Text Java - A Java implementation of Twitter's text processing library.
  • MALLET - A Java-based package for statistical natural language processing, document classification, clustering, topic modeling, information extraction, and other machine learning applications to text.
  • OpenNLP - a machine learning based toolkit for the processing of natural language text.
  • LingPipe - A tool kit for processing text using computational linguistics.
  • ClearTKstars129 - ClearTK provides a framework for developing statistical natural language processing (NLP) components in Java and is built on top of Apache UIMA. [Deprecated]
  • Apache cTAKES - Apache Clinical Text Analysis and Knowledge Extraction System (cTAKES) is an open-source natural language processing system for information extraction from electronic medical record clinical free-text.
  • NLP4Jstars145 - The NLP4J project provides software and resources for natural language processing. The project started at the Center for Computational Language and EducAtion Research, and is currently developed by the Center for Language and Information Research at Emory University. [Deprecated]
  • CogcompNLPstars453 - This project collects a number of core libraries for Natural Language Processing (NLP) developed in the University of Illinois' Cognitive Computation Group, for example illinois-core-utilities which provides a set of NLP-friendly data structures and a number of NLP-related utilities that support writing NLP applications, running experiments, etc, illinois-edison a library for feature extraction from illinois-core-utilities data structures and many other packages.

General-Purpose Machine Learning

  • aerosolvestars4.8k - A machine learning library by Airbnb designed from the ground up to be human friendly.
  • AMIDST Toolbox - A Java Toolbox for Scalable Probabilistic Machine Learning.
  • Chips-n-Salsastars24 - A Java library for genetic algorithms, evolutionary computation, and stochastic local search, with a focus on self-adaptation / self-tuning, as well as parallel execution.
  • Datumboxstars1.1k - Machine Learning framework for rapid development of Machine Learning and Statistical applications.
  • ELKI - Java toolkit for data mining. (unsupervised: clustering, outlier detection etc.)
  • Encogstars739 - An advanced neural network and machine learning framework. Encog contains classes to create a wide variety of networks, as well as support classes to normalize and process data for these neural networks. Encog trainings using multithreaded resilient propagation. Encog can also make use of a GPU to further speed processing time. A GUI based workbench is also provided to help model and train neural networks.
  • FlinkML in Apache Flink - Distributed machine learning library in Flink.
  • H2Ostars6k - ML engine that supports distributed learning on Hadoop, Spark or your laptop via APIs in R, Python, Scala, REST/JSON.
  • htm.javastars302 - General Machine Learning library using Numenta’s Cortical Learning Algorithm.
  • liblinear-javastars306 - Java version of liblinear.
  • Mahoutstars2k - Distributed machine learning.
  • Meka - An open source implementation of methods for multi-label classification and evaluation (extension to Weka).
  • MLlib in Apache Spark - Distributed machine learning library in Spark.
  • Hydrosphere Miststars318 - a service for deployment Apache Spark MLLib machine learning models as realtime, batch or reactive web services.
  • Neuroph - Neuroph is lightweight Java neural network framework.
  • ORYXstars1.8k - Lambda Architecture Framework using Apache Spark and Apache Kafka with a specialization for real-time large-scale machine learning.
  • Samoa SAMOA is a framework that includes distributed machine learning for data streams with an interface to plug-in different stream processing platforms.
  • RankLib - RankLib is a library of learning to rank algorithms. [Deprecated]
  • rapaiostars63 - statistics, data mining and machine learning toolbox in Java.
  • RapidMiner - RapidMiner integration into Java code.
  • Stanford Classifier - A classifier is a machine learning tool that will take data items and place them into one of k classes.
  • Smile - Statistical Machine Intelligence & Learning Engine.
  • SystemMLstars923 - flexible, scalable machine learning (ML) language.
  • Weka - Weka is a collection of machine learning algorithms for data mining tasks.
  • LBJava - Learning Based Java is a modeling language for the rapid development of software systems, offers a convenient, declarative syntax for classifier and constraint definition directly in terms of the objects in the programmer's application.
  • knn-java-librarystars3 - Just a simple implementation of K-Nearest Neighbors algorithm using with a bunch of similarity measures.

Speech Recognition

  • CMU Sphinx - Open Source Toolkit For Speech Recognition purely based on Java speech recognition library.

Data Analysis / Data Visualization

  • Flink - Open source platform for distributed stream and batch data processing.
  • Hadoopstars13k - Hadoop/HDFS.
  • Onyxstars2k - Distributed, masterless, high performance, fault tolerant data processing. Written entirely in Clojure.
  • Sparkstars34k - Spark is a fast and general engine for large-scale data processing.
  • Storm - Storm is a distributed realtime computation system.
  • Impalastars20 - Real-time Query for Hadoop.
  • DataMelt - Mathematics software for numeric computation, statistics, symbolic calculations, data analysis and data visualization.
  • Dr. Mich