Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

Deep learning PyTorch library for time series forecasting, classification, and anomaly detection (originally for flood forecasting).

🐍 Geometric Computer Vision Library for Spatial AI

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

Live training loss plot in Jupyter Notebook for Keras, PyTorch and others

Fast EMD for Python: a wrapper for Pele and Werman's C++ implementation of the Earth Mover's Distance metric