Top 20 Python AI and Machine Learning Open Source Projects
Getting into Machine Learning and AI is not an easy task, but is a critical part of data science programs. Many aspiring professionals and enthusiasts find it hard to establish a proper path into the field, given the enormous amount of resources available today. The field is evolving constantly and it is crucial that we keep up with the pace of this rapid development. In order to cope with this overwhelming speed of evolution and innovation, a good way to stay updated and knowledgeable on the advances of ML, is to engage with the community by contributing to the many open-source projects and tools that are used daily by advanced professionals.
Here we update the information and examine the trends since our previous post Top 20 Python Machine Learning Open Source Projects (Nov 2016).
Tensorflow has moved to the first place with triple-digit growth in contributors. Scikit-learn dropped to 2nd place, but still has a very large base of contributors.
Compared to 2016, the projects with the fastest growth in number of contributors were
- TensorFlow, 169% up, from 493 to 1324 contributors
- Deap, 86% up, from 21 to 39 contributors
- Chainer, 83% up, from 84 to 154 contributors
- Gensim, 81% up, from 145 to 262 contributors
- Neon, 66% up, from 47 to 78 contributors
- Nilearn, 50% up, from 46 to 69 contributors
Also new in 2018:
- Keras, 629 contributors
- PyTorch, 399 contributors
Size is proportional to the number of contributors, and color represents to the change in the number of contributors – red is higher, blue is lower. Snowflake shape is for Deep Learning projects, round for other projects.
We see that Deep Learning projects like TensorFlow, Theano, and Caffe are among the most popular.
The list below gives projects in descending order based on the number of contributors on Github. The change in number of contributors is versus 2016 KDnuggets Post on Top 20 Python Machine Learning Open Source Projects.
We hope you enjoy going through the documentation pages of each of these to start collaborating and learning the ways of Machine Learning using Python.
- TensorFlow was originally developed by researchers and engineers working on the Google Brain Team within Google’s Machine Intelligence research organization. The system is designed to facilitate research in machine learning, and to make it quick and easy to transition from research prototype to production system.
Contributors: 1324 (168% up), Commits: 28476, Stars: 92359. Github URL: Tensorflow
- Scikit-learn is simple and efficient tools for data mining and data analysis, accessible to everybody, and reusable in various context, built on NumPy, SciPy, and matplotlib, open source, commercially usable – BSD license.
Contributors: 1019 (39% up), Commits: 22575, Github URL: Scikit-learn
- Keras, a high-level neural networks API, written in Python and capable of running on top of TensorFlow, CNTK, or Theano.
Contributors: 629 (new), Commits: 4371, Github URL: Keras
- PyTorch, Tensors and Dynamic neural networks in Python with strong GPU acceleration.
Contributors: 399 (new), Commits: 6458, Github URL: pytorch
- Theano allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays efficiently.
Contributors: 327 (24% up), Commits: 27931, Github URL: Theano
- Gensim is a free Python library with features such as scalable statistical semantics, analyze plain-text documents for semantic structure, retrieve semantically similar documents.
Contributors: 262 (81% up), Commits: 3549, Github URL: Gensim
- Caffe is a deep learning framework made with expression, speed, and modularity in mind. It is developed by the Berkeley Vision and Learning Center (BVLC) and community contributors.
Contributors: 260 (21% up), Commits: 4099, Github URL: Caffe
- Chainer is a Python-based, standalone open source framework for deep learning models. Chainer provides a flexible, intuitive, and high performance means of implementing a full range of deep learning models, including state-of-the-art models such as recurrent neural networks and variational auto-encoders.
Contributors: 154 (84% up), Commits: 12613, Github URL: Chainer
- Statsmodels is a Python module that allows users to explore data, estimate statistical models, and perform statistical tests. An extensive list of descriptive statistics, statistical tests, plotting functions, and result statistics are available for different types of data and each estimator.
Contributors: 144 (33% up), Commits: 9729, Github URL: Statsmodels
- Shogun is Machine learning toolbox which provides a wide range of unified and efficient Machine Learning (ML) methods. The toolbox seamlessly allows to easily combine multiple data representations, algorithm classes, and general purpose tools.
Contributors: 139 (32% up), Commits: 16362, Github URL: Shogun
- Pylearn2 is a machine learning library. Most of its functionality is built on top of Theano. This means you can write Pylearn2 plugins (new models, algorithms, etc) using mathematical expressions, and Theano will optimize and stabilize those expressions for you, and compile them to a backend of your choice (CPU or GPU).
Contributors: 119 (3.5% up), Commits: 7119, Github URL: Pylearn2
- NuPIC is an open source project based on a theory of neocortex called Hierarchical Temporal Memory (HTM). Parts of HTM theory have been implemented, tested, and used in applications, and other parts of HTM theory are still being developed.
Contributors: 85 (12% up), Commits: 6588, Github URL: NuPIC
- Neon is Nervana’s Python-based deep learning library. It provides ease of use while delivering the highest performance.
Contributors: 78 (66% up), Commits: 1112, Github URL: Neon
- Nilearn is a Python module for fast and easy statistical learning on NeuroImaging data. It leverages the scikit-learn Python toolbox for multivariate statistics with applications such as predictive modelling, classification, decoding, or connectivity analysis.
Contributors: 69 (50% up), Commits: 6198, Github URL: Nilearn
- Orange3 is open source machine learning and data visualization for novice and expert. Interactive data analysis workflows with a large toolbox.
Contributors: 53 (33% up), Commits: 8915, Github URL: Orange3
- Pymc is a python module that implements Bayesian statistical models and fitting algorithms, including Markov chain Monte Carlo. Its flexibility and extensibility make it applicable to a large suite of problems.
Contributors: 39 (5.4% up), Commits: 2721, Github URL: Pymc
- Deap is a novel evolutionary computation framework for rapid prototyping and testing of ideas. It seeks to make algorithms explicit and data structures transparent. It works in perfect harmony with parallelisation mechanism such as multiprocessing and SCOOP.
Contributors: 39 (86% up), Commits: 1960, Github URL: Deap
- Annoy (Approximate Nearest Neighbors Oh Yeah) is a C++ library with Python bindings to search for points in space that are close to a given query point. It also creates large read-only file-based data structures that are mapped into memory so that many processes may share the same data.
Contributors: 35 (46% up), Commits: 527, Github URL: Annoy
- PyBrain is a modular Machine Learning Library for Python. Its goal is to offer flexible, easy-to-use yet still powerful algorithms for Machine Learning Tasks and a variety of predefined environments to test and compare your algorithms.
Contributors: 32 (3% up), Commits: 992, Github URL: PyBrain
- Fuel is a data pipeline framework which provides your machine learning models with the data they need. It is planned to be used by both the Blocks and Pylearn2 neural network libraries.
Contributors: 32 (10% up), Commits: 1116, Github URL: Fuel
The contributor and commit numbers were recorded in February 2018.
Editor’s note: This was originally posted on KDNuggets, and has been reposted with permission. Author Ilan Reinstein is a physicist and data scientist.
Free Data Science Resources
Sign up for free to get our weekly newsletter with data science, Python, R, and SQL resource links. Plus, you get access to our free, interactive online courses!
Data Scientist at NYU Langone Medical Center. MS Applied Physics – NYU Tandon School of Engineering BS Physics – Universidad de Los Andes (Colombia)