jax.typeof

jax.typeof(x, /)

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