flax

bb3c31ac - Deduplicate `Variable` nodes in `iter_graph` and eliminate recursion.