jax.numpy.trace =============== .. currentmodule:: jax.numpy .. autofunction:: trace