flashtorch (pypi): Visualization toolkit for neural networks in PyTorch
Keywords: cnn|deep-learning|explainability|interpretability|machine-learning|neural-networks|pytorch|visualization