jax.lax.cos =========== .. currentmodule:: jax.lax .. autofunction:: cos