jax.lax.imag ============ .. currentmodule:: jax.lax .. autofunction:: imag