A Python package for segmenting geospatial data with the SAM
Time series forecasting with PyTorch
A ranked list of awesome machine learning Python libraries
End-to-End Library for Continual Learning based on PyTorch
Decentralized deep learning in PyTorch. Built to train models
Medical imaging toolkit for deep learning
DeepVariant is an analysis pipeline that uses a deep neural networks
Making large AI models cheaper, faster and more accessible
Graph Neural Network Library for PyTorch
Models and examples built with TensorFlow
Package of deep-learning based CTR models
Open deep learning compiler stack for cpu, gpu, etc.
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
ktrain is a Python library that makes deep learning AI more accessible
A Python library for audio data augmentation
PyTorch extensions for fast R&D prototyping and Kaggle farming
Create HTML profiling reports from pandas DataFrame objects
The most intuitive, flexible, way for researchers to build models
A fast library for AutoML and tuning
Implementation of Imagen, Google's Text-to-Image Neural Network
A scikit-learn compatible neural network library that wraps PyTorch
Hummingbird compiles trained ML models into tensor computation
Machine learning metrics for distributed, scalable PyTorch application
A library of extension and helper modules for Python's data analysis
A Python Package to Tackle the Curse of Imbalanced Datasets in ML