Framework | Description |
TensorFlow | An Open Source Machine Learning Framework for Everyone. |
scikit-learn | Machine learning in Python. |
PyTorch | Tensors and Dynamic neural networks in Python with strong GPU support. |
Keras | Deep Learning for humans. |
StatsModels | Statistical modeling and econometrics in Python. |
XGBoost | Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT, or GBM). |
jax | Composable transformations of Python+NumPy programs: differentiate, vectorize, just-in-time compilation to GPU/TPU. |
LightGBM | A fast, distributed, high-performance gradient boosting framework. |
pytorch-lightning | Deep learning framework to train, deploy, and ship AI models. |
PaddlePaddle | Parallel Distributed Deep Learning: Machine Learning Framework. |
Fastai | A deep learning library. |
Jina | Build multimodal AI services via cloud-native technologies. |
PySpark | Apache Spark Python API. |
MXNet | Lightweight, portable, flexible distributed/mobile deep learning framework. |
Catboost | A fast, scalable, high-performance Gradient Boosting on Decision Trees. |
Flax | A neural network library for JAX designed for flexibility. |
Thinc | A refreshing functional take on deep learning, compatible with your favorite libraries. |
Vowpal Wabbit | A machine learning system that pushes the frontier of machine learning. |
einops | Deep learning operations reinvented (for PyTorch, TensorFlow, JAX, and NumPy). |
ivy | The Unified Machine Learning Framework. |
Ludwig | Data-centric declarative deep learning framework. |
tensorpack | A neural net training interface on TensorFlow, with a focus on speed + flexibility. |
Chainer | A flexible framework of neural networks for deep learning. |
PyFlink | Apache Flink Python API. |
Sonnet | TensorFlow-based neural network library. |
skorch | A scikit-learn compatible neural network library that wraps PyTorch. |
Ignite | A high-level library to help with training and evaluating neural networks. |
Haiku | A JAX-based neural network library. |
ktrain | A Python library that makes deep learning and AI more accessible. |
tensorflow-upstream | TensorFlow ROCm port. |
Neural Network Libraries | Neural Network Libraries. |
Geomstats | Computations and statistics on manifolds with geometric structures. |
DyNet | The Dynamic Neural Network Toolkit. |
Towhee | A framework dedicated to making neural data easy and efficient. |
Neural Tangents | Fast and Easy Infinite Neural Networks in Python. |
xLearn | High-performance, easy-to-use, and scalable machine learning (ML) library. |
fklearn | Functional Machine Learning library. |
NeuPy | A TensorFlow-based Python library for prototyping and building neural networks. |
mace | A deep learning inference framework optimized for mobile devices. |
ThunderSVM | A fast SVM library on GPUs and CPUs. |
NeoML | A machine learning framework for both deep learning and traditional models. |
chefboost | A lightweight decision tree framework supporting regular algorithms. |
elegy | A high-level API for deep learning in JAX. |
ThunderGBM | Fast GBDTs and Random Forests on GPUs. |
Library | Description |
Matplotlib | Plotting with Python |
Seaborn | Statistical data visualization in Python |
Bokeh | Interactive data visualization in the browser, from Python |
Plotly | Interactive graphing library for Python |
Altair | Declarative statistical visualization library for Python |
dash | Data apps and dashboards for Python |
pyecharts | Python Echarts plotting library |
plotnine | Grammar of graphics for Python |
PyQtGraph | Fast data visualization and GUI tools for scientific/engineering |
FiftyOne | Visualize, create, and debug image and video datasets |
PyVista | 3D plotting and mesh analysis through a streamlined interface |
UMAP | Uniform Manifold Approximation and Projection |
HoloViews | Data visualization library that makes data visualize itself |
Graphviz | Simple Python interface for Graphviz |
datashader | Render large datasets quickly and accurately |
D-Tale | Visualizer for pandas data structures |
bqplot | Plotting library for IPython/Jupyter notebooks |
mpld3 | D3 renderings of Matplotlib graphics |
hvPlot | High-level plotting API for pandas, dask, xarray, and networkx |
wordcloud | Word cloud generator in Python |
missingno | Missing data visualization module for Python |
Facets Overview | Visualizations for machine learning datasets |
data-validation | Library for exploring and validating machine learning |
Perspective | Data visualization and analytics component |
openTSNE | Extensible, parallel implementations of t-SNE |
lets-plot | Open-source plotting library for statistical data |
Chartify | Library to create interactive visualizations easily |
Plotly-Resampler | Visualize large time series data with Plotly.py |
AutoViz | Automatically visualize any dataset with a single line of code |
HiPlot | Tool to visualize high-dimensional data |
Pandas-Bokeh | Bokeh plotting backend for Pandas and GeoPandas |
Multicore-TSNE | Parallel t-SNE implementation with Python and Torch |
python-ternary | Ternary plotting library for Python with Matplotlib |
Sweetviz | Visualize and compare datasets easily |
Popmon | Monitor the stability of Pandas or Spark dataframes |
vega | IPython/Jupyter notebook module for Vega and Vega-Lite |
PyWaffle | Make waffle charts in Python |