jax.numpy.identity ================== .. currentmodule:: jax.numpy .. autofunction:: identity