as_strong_pytree()

liesel.goose.pytree.as_strong_pytree(pytree)

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)