Contents

jax.flatten_util package

Warning

This page was created from a pull request (#9655).

jax.flatten_util package¶

List of Functions¶

ravel_pytree(pytree)

Ravel (i.e.

previous

jax.experimental.checkify.check_error

next

jax.flatten_util.ravel_pytree