Awesome Python Data Science

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

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.

The open source AI engineering platform for agents, LLMs, and ML models. MLflow enables teams of all sizes to debug, evaluate, monitor, and optimize production-quality AI applications while controlling costs and managing access to models and data.

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.

Train transformer language models with reinforcement learning.

Network Analysis in Python

🦉 Data Versioning and ML Experiments

🎨 Python Echarts Plotting Library

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

A hyperparameter optimization framework

Image augmentation for machine learning experiments.

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.

1 Line of code data quality profiling & exploratory data analysis for Pandas and Spark DataFrames.

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

Statsmodels: statistical modeling and econometrics in Python

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.

An elegant PyTorch deep reinforcement learning library.

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

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

A unified framework for machine learning with time series

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

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

Bayesian Modeling and Probabilistic Programming in Python

cuDF - GPU DataFrame Library

Automatic extraction of relevant features from time series:

AutoML library for deep learning

Deep universal probabilistic programming with Python and PyTorch

A Python implementation of global optimization with gaussian processes.

Python library for audio and music analysis

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

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

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

Efficiently computes derivatives of NumPy code.

Deep Learning and Reinforcement Learning Library for Scientists and Engineers

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