liesel.goose.pytree.as_strong_pytree
liesel.goose.pytree.as_strong_pytree#
- liesel.goose.pytree.as_strong_pytree(pytree)[source]#
Converts every leaf in a pytree to a non-weak
jax.numpy.DeviceArray
.See https://jax.readthedocs.io/en/latest/type_promotion.html.
- Return type
TypeVar
(T
)