jax.lax.atan ============ .. currentmodule:: jax.lax .. autofunction:: atan