I agree visualizing neural networks is challenging. The flexibility and consequences of network connections is also confusing. If you decide to go down that route again, the best work I've seen by far is the recent one by Stephen Wolfram: https://writings.stephenwolfram.com/2024/08/whats-really-goi...