quantax.utils.tree_fully_flatten#

quantax.utils.tree_fully_flatten(tree: PyTree) Array#

Return the array given by jax.flatten_util.ravel_pytree