I've been playing around with Keras's built-in visualizations tool for a while and I feel like it could be improved a lot. The current visualization system doesn't provide much information "at a glance" (meaning it takes quite a bit of analysis to understand what's going on) and isn't super aesthetically pleasing.

I'm opening this issue to see if an improved visualization system is something that the Keras team would be open to. I'd be happy to make small improvements to the existing system, but I feel like it would be better to add a new visualization system (or submodule) for highly customizable visual architecture generation. I could bring some (or most) of the key features from visualkeras, a repository I maintain, into the src/visualization module.

Since we'd isolate the dependencies of the new features from Keras as a whole (like what was done for the last system), we wouldn't be adding any bloat through more dependencies.

Below are a few examples of what the visualizations I'd add may look like