GitHub repository: vainaixr/flashtorch
Primary language: Python
Description: Visualization toolkit for neural networks in PyTorch! Demo -->