Use weakref

Use weakref for the backward references (node from output, node from input, etc) to avoid blocking the garbage collector.

This should be discussed due to possible overhead.