jax.lax.sign ============ .. currentmodule:: jax.lax .. autofunction:: sign