[Project] How to create visualizations for "complex" networks?

For a presentation I need to make my own visualization of a graph neural network with encoders and other additional parts. I would like to get something like this:

​

https://preview.redd.it/yi1erwam1t091.png?width=868&format=png&auto=webp&s=adabbe066182b2065b835ea50e8fc83c3574910a

Does anyone know a convenient way to do this?

machinelearning networks project