10 Best Python Machine Learning Libraries

List hand-picked by Openbase Experts
Learn More

Keras

Deep Learning for humans

License Icon
License: MIT
User Rating
4.5/ 5
2
Top Feedback
2Easy to Use
1Great Documentation
1Performant
GitHub Stars
53K
Weekly Downloads
0
Last Commit
1mo ago

scikit-learn

scikit-learn: machine learning in Python

License Icon
License: new BSD
User Rating
4.0/ 5
1
Top Feedback
N/A
GitHub Stars
48K
Weekly Downloads
0
Last Commit
2mos ago

tensorflow-gpu

An Open Source Machine Learning Framework for Everyone

License Icon
License: Apache 2.0
User RatingN/A
Top Feedback
N/A
GitHub Stars
161K
Weekly Downloads
0
Last Commit
1mo ago

torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

License Icon
License: BSD-3
User RatingN/A
Top Feedback
N/A
GitHub Stars
52K
Weekly Downloads
0
Last Commit
2mos ago

ray

An open source framework that provides a simple, universal API for building distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library.

License Icon
License: Apache 2.0
User RatingN/A
Top Feedback
N/A
GitHub Stars
18K
Weekly Downloads
0
Last Commit
2mos ago

pyspark

Apache Spark - A unified analytics engine for large-scale data processing

License Icon
License: http://www.apache.org/licenses/LICENSE-2.0
User RatingN/A
Top Feedback
N/A
GitHub Stars
31K
Weekly Downloads
0
Last Commit
2mos ago

MindsDB

Predictive AI layer for existing databases.

License Icon
License: GPL-3.0
User RatingN/A
Top Feedback
N/A
GitHub Stars
4K
Weekly Downloads
0
Last Commit
1mo ago

h2o

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

License Icon
License: Apache v2
User RatingN/A
Top Feedback
N/A
GitHub Stars
6K
Weekly Downloads
0
Last Commit
1mo ago

xgboost

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

License Icon
License: Apache-2.0
User RatingN/A
Top Feedback
N/A
GitHub Stars
22K
Weekly Downloads
0
Last Commit
1mo ago

mxnet

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

License Icon
License: Apache 2.0
User RatingN/A
Top Feedback
N/A
GitHub Stars
20K
Weekly Downloads
0
Last Commit
2mos ago

gym

A toolkit for developing and comparing reinforcement learning algorithms.

License Icon
License: Unknown
User RatingN/A
Top Feedback
N/A
GitHub Stars
26K
Weekly Downloads
0
Last Commit
2mos ago

albumentations

Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: https://albumentations.ai/docs/ Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

License Icon
License: MIT
User RatingN/A
Top Feedback
N/A
GitHub Stars
9K
Weekly Downloads
0
Last Commit
2mos ago
lea

learnergy

💡 Learnergy is a Python library for energy-based machine learning models.

License Icon
License: GPL-3.0
User RatingN/A
Top Feedback
N/A
GitHub Stars
36
Weekly Downloads
0
Last Commit
2mos ago
tl

torch-lucent

Lucid library adapted for PyTorch

License Icon
License: Apache License 2.0
User RatingN/A
Top Feedback
N/A
GitHub Stars
394
Weekly Downloads
0
Last Commit
6mos ago
fr

face-recognition

The world's simplest facial recognition api for Python and the command line

License Icon
License: MIT license
User RatingN/A
Top Feedback
N/A
GitHub Stars
42K
Weekly Downloads
0
Last Commit
7mos ago
ser

SerpentAI

Game Agent Development Kit. Helping you create AIs / Bots to play any game you own!

License Icon
License: MIT
User RatingN/A
Top Feedback
N/A
GitHub Stars
N/A
Weekly Downloads
0
Last Commit
N/A
neu

neupy

NeuPy is a Python library for Artificial Neural Networks and Deep Learning.

License Icon
License: MIT
User RatingN/A
Top Feedback
N/A
GitHub Stars
N/A
Weekly Downloads
0
Last Commit
N/A
vp

vowpal_porpoise

lightweight python wrapper for vowpal wabbit

License Icon
License: BSD
User RatingN/A
Top Feedback
N/A
GitHub Stars
161
Weekly Downloads
0
Last Commit
2yrs ago
ns

neural-style

PyTorch implementation of neural style transfer algorithm

License Icon
License: Unknown
User RatingN/A
Top Feedback
N/A
GitHub Stars
607
Weekly Downloads
0
Last Commit
1yr ago

nupic

Numenta Platform for Intelligent Computing is an implementation of Hierarchical Temporal Memory (HTM), a theory of intelligence based strictly on the neuroscience of the neocortex.

License Icon
License: Unknown
User RatingN/A
Top Feedback
N/A
GitHub Stars
6K
Weekly Downloads
0
Last Commit
2yrs ago