Awesome Python Data Science

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Extremely fast Query Engine for DataFrames, written in Rust

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

🎨 Python Echarts Plotting Library

A hyperparameter optimization framework

Statistical data visualization in Python

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

Fast and Accurate ML in 3 Lines of Code

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 python library for user-friendly forecasting and anomaly detection on time series.

Deep universal probabilistic programming with Python and PyTorch

A Python implementation of global optimization with gaussian processes.

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

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

Distributed Asynchronous Hyperparameter Optimization in Python

Efficiently computes derivatives of NumPy code.

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

Python tools for geographic data

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

Lightning ⚡️ fast forecasting with statistical and econometric models.

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

Scalable and user friendly neural :brain: forecasting algorithms.

N-D labeled arrays and datasets in Python

A Python package for interactive geospatial analysis and visualization with Google Earth Engine.

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

With Holoviews, your data visualizes itself.

Apache Hamilton helps data scientists and engineers define testable, modular, self-documenting dataflows, that encode lineage/tracing and metadata. Runs and scales everywhere python does.

Fast numerical array expression evaluator for Python, NumPy, Pandas, PyTables and more

A high performance Python graph library implemented in Rust.

Lightweight and extensible compatibility layer between dataframe libraries!

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

Scalable machine 🤖 learning for time series forecasting.

:chart_with_upwards_trend: Adaptive: parallel active learning of mathematical functions

Fast NumPy array functions written in C

The Classical Language Toolkit

Bias Auditing & Fair ML Toolkit

Easy pipelines for pandas DataFrames.

Solve automatic numerical differentiation problems in one or more variables.

Relevance Vector Machine implementation using the scikit-learn API.

BatchFlow helps you conveniently work with random or sequential batches of your data and define data processing and machine learning workflows even for datasets that do not fit into memory.

Python histogram library - histograms as updateable, fully semantic objects with visualization tools. [P]ython [HYST]ograms.