Awesome Python Data Science

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

A game theoretic approach to explain the output of any machine learning model.

Graph Neural Network Library for PyTorch

matplotlib: plotting with Python

Open standard for machine learning interoperability

Tool for producing high quality forecasts for time series data that has multiple seasonality with linear or non-linear growth.

🎨 Python Echarts Plotting Library

NLTK Source

A very simple framework for state-of-the-art Natural Language Processing (NLP)

Statistical data visualization in Python

Parallel computing with task scheduling

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

A standard API for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

Lime: Explaining the predictions of any machine learning classifier

Low-code framework for building custom LLMs, neural networks, and other AI models

Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

Always know what to expect from your data.

Statsmodels: statistical modeling and econometrics in Python

An elegant PyTorch deep reinforcement learning library.

Modin: Scale your Pandas workflows by changing a single line of code

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

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

A unified framework for machine learning with time series

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

cuDF - GPU DataFrame Library

Bayesian Modeling and Probabilistic Programming in Python

AutoML library for deep learning

Automatic extraction of relevant features from time series:

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

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.

An open source python library for automated feature engineering

Distributed Asynchronous Hyperparameter Optimization in Python

Deep Learning and Reinforcement Learning Library for Scientists and Engineers

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

Image processing in Python

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

cuML - RAPIDS Machine Learning Library

Python tools for geographic data

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

Image augmentation library in Python for machine learning.

A Python implementation of LightFM, a hybrid recommendation algorithm.

Lightning ⚡️ fast forecasting with statistical and econometric models.

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

A light-weight, flexible, and expressive statistical data testing library