[WIP] Quickly generate architecture diagrams of PyTorch or Keras models in a notebook. - View it on GitHub