Awesome Python Data Science

Fast and Accurate ML in 3 Lines of Code

A Python Automated Machine Learning tool that optimizes machine learning pipelines using genetic programming.

TensorFlow-based neural network library

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Open-source, low-code AutoML platform for Python. PyCaret 4.0: sklearn-native engine + React control plane.

A unified framework for machine learning with time series

cuDF - GPU DataFrame Library

Bayesian Modeling and Probabilistic Programming in Python

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

A python library for user-friendly forecasting and anomaly detection on time series.

AutoML library for deep learning

Automatic extraction of relevant features from time series:

Deep universal probabilistic programming with Python and PyTorch

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

Web mining module for Python, with tools for scraping, natural language processing, machine learning, network analysis and visualization.

A Python implementation of global optimization with gaussian processes.

Out-of-Core hybrid Apache Arrow/NumPy DataFrame for Python, ML, visualization and exploration of big tabular data at a billion rows per second 🚀

Python library for audio and music analysis

Automated Machine Learning with scikit-learn

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

An open source python library for automated feature engineering

Distributed Asynchronous Hyperparameter Optimization in Python

Evidently is ​​an open-source ML and LLM observability framework. Evaluate, test, and monitor any AI-powered system or data pipeline. From tabular data to Gen AI. 100+ metrics.

Efficiently computes derivatives of NumPy code.

Qiskit is an open-source SDK for working with quantum computers at the level of extended quantum circuits, operators, and primitives.

Deep Learning and Reinforcement Learning Library for Scientists and Engineers

Flax is a neural network library for JAX that is designed for flexibility.

A Python Package to Tackle the Curse of Imbalanced Datasets in Machine Learning

Fit interpretable models. Explain blackbox machine learning.

A Python scikit for building and analyzing recommender systems

Genetic Algorithm, Particle Swarm Optimization, Simulated Annealing, Ant Colony Optimization Algorithm,Immune Algorithm, Artificial Fish Swarm Algorithm, Differential Evolution and TSP(Traveling salesman)

Image processing in Python

Distributed Evolutionary Algorithms in Python

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

A scikit-learn compatible neural network library that wraps PyTorch

A system for quickly generating training data with weak supervision

Uplift modeling and causal inference with machine learning algorithms

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

mlpack: a fast, header-only C++ machine learning library

Deep Reinforcement Learning for Keras.

Build Graph Nets in Tensorflow

cuML - RAPIDS Machine Learning Library

A library of extension and helper modules for Python's data analysis and machine learning libraries.

Python tools for geographic data

Image augmentation library in Python for machine learning.

A Python implementation of LightFM, a hybrid recommendation algorithm.

Python framework for creating, editing, and running Noisy Intermediate-Scale Quantum (NISQ) circuits.

Lightning ⚡️ fast forecasting with statistical and econometric models.

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.