beautiful-tensors (pypi): Turn tensor operations into aesthetic diagrams. Currently supports PyTorch (Keras/TensorFlow coming soon).
Keywords: tensors|beautiful|diagram|pytorch|tensorflow|matrix|operations|pictures|keras