jax.typeof — JAX documentation (original) (raw)

jax.typeof#

jax.typeof(x, /)[source]#

Return the JAX type (i.e. AbstractValue) of the input.

Raises a TypeError if x is not a valid JAX type.

Parameters:

x (Any)

Return type:

Any