jax.flatten_util package¶

List of Functions¶

ravel_pytree(pytree)

Ravel (i.e.