Bug in ReadMe graph

The tags of the graph do not match the code, e.g, i2h

I am not sure to understand what you mean by that? The graph looks ok by me.
Keep in mind that Variables that are in the middle of the graph exist only while associated to a python variable but the computational graph itself contains only Functions, the Variables between them are not kept as is.

I think @Huaiwen_Zhang is pointing out that i2h and h2h seem to be mixed-up.

Ho right my bad !
This thing is not that easy to read as it keeps moving.

Not sure who generated the gif though, @smth might know?