flax

3f13b23b - Merge pull request #1099 from jheek:add-pytreenode-docs