Visualkeras, a Python package, simplifies the visualization of Keras (standalone or in TensorFlow) neural network architectures. It provides flexible styling options, supporting layered style for CNNs and graph style for various models, including feed-forward networks.
https://ashokri.com/demonstrate-layer-dimension-in-visualcross/