jax.numpy.square ================ .. currentmodule:: jax.numpy .. autofunction:: square