Model visualisation
We might integrate better visualisations of the model. Here is a repo wich summarises different tools (we have to evaluate which ones can read Kras/TF model structures).
https://github.com/ashishpatel26/Tools-to-Design-or-Visualize-Architecture-of-Neural-Network