jax.lax.while\_loop
===================

.. currentmodule:: jax.lax

.. autofunction:: while_loop