This is great! Although if your goal is to make this more approachable, may I suggest that you include a diagram in your documentation, explaining how the layer

This is great! Although if your goal is to make this more approachable, may I suggest that you include a diagram in your documentation, explaining how the layers look and how they are interconnected? I studied ANNs in university, but after 10 years, even I have trouble recalling some details.