The kwarray module implements a small set of pure-python extensions to numpy and torch.

array numpy torch