jax.lax.exp =========== .. currentmodule:: jax.lax .. autofunction:: exp