jax.lax.square
==============

.. currentmodule:: jax.lax

.. autofunction:: square