Nodes can generate PyTrees with new build_pytree() function which will generate a pytree dictionary for the current tree of nodes. The same dictionary can be passed through the sampling functions as given_args to override any of the internal settings.
Nodes can generate PyTrees with new
build_pytree()function which will generate a pytree dictionary for the current tree of nodes. The same dictionary can be passed through the sampling functions asgiven_argsto override any of the internal settings.