Closed ASEM000 closed 1 year ago
Specifically, show why it can be helpful compared to tree-specific flatten rules foundflax.struct.PyTreeNode/ equinox.Module and other PyTree based libraries.
flax.struct.PyTreeNode
equinox.Module
PyTree
Specifically, show why it can be helpful compared to tree-specific flatten rules found
flax.struct.PyTreeNode
/equinox.Module
and otherPyTree
based libraries.