๐
Working from home
- Palaiseau, France
Pinned Loading
-
utility apply reduce function to a p...
utility apply reduce function to a pytree (or a pytree of pytree) 1import numpy as np
23from jax import numpy as jnp
4from jax.lib import pytree
5
Something went wrong, please refresh the page to try again.
If the problem persists, check the GitHub status page or contact support.
If the problem persists, check the GitHub status page or contact support.