It would be really awesome if someone could package this into a Latex package. Then you could directly define your neural net inside of Latex instead of having to generate the Tex from python.

Otherwise, these graphics are wonderful to look at.