The kwarray module implements a small set of pure-python extensions to numpy and torch.
https://gitlab.kitware.com/computer-vision/kwarray
stable