Awesome Python Data Science

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

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!

A toolkit for developing and comparing reinforcement learning algorithms.

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.

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.

Datasets, Transforms and Models specific to Computer Vision

Network Analysis in Python

🎨 Python Echarts Plotting Library

🦉 Data Versioning and ML Experiments

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

Image augmentation for machine learning experiments.

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

NLTK Source

A hyperparameter optimization framework

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

Python package built to ease deep learning on graph, on top of existing DL frameworks.

Statistical data visualization in Python

Parallel computing with task scheduling

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

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)

NumPy & SciPy for GPU

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.

Fast and Accurate ML in 3 Lines of Code

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)

TensorFlow-based neural network library

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

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.

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

Deep universal probabilistic programming with Python and PyTorch

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