A ranked list of awesome machine learning Python libraries
Open source machine learning framework
Visualizer for neural network, deep learning, machine learning models
A framework for real-life data science
A unified framework for scalable computing
Jittor is a high-performance deep learning framework
Medical imaging toolkit for deep learning
AutoML library for deep learning
Graph Neural Network Library for PyTorch
Python package built to ease deep learning on graph
Petastorm library enables single machine or distributed training
Library for OCR-related tasks powered by Deep Learning
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
A python library for self-supervised learning on images
Fast image augmentation library and an easy-to-use wrapper
Deep learning optimization library making distributed training easy
ktrain is a Python library that makes deep learning AI more accessible
Deep learning library
High-level, high-performance dynamic language for technical computing
A python library for easy manipulation and forecasting of time series
Hub of ready-to-use datasets for ML models
Geometric deep learning extension library for PyTorch
Open source framework for deep learning satellite and aerial imagery
Making large AI models cheaper, faster and more accessible
Simple and distributed Machine Learning