jax.flatten_util package

List of Functions

ravel_pytree(pytree)

Ravel (i.e.