pip install quiver_engine==0.1.4.1.4

Interactive per-layer visualization for convents in keras

Source