oryx.core.trace_util.get_shaped_aval

Converts a JAX value type into a shaped abstract value.

Main aliases

oryx.experimental.matching.jax_rewrite.trace_util.get_shaped_aval