jax.lax.top\_k
==============

.. currentmodule:: jax.lax

.. autofunction:: top_k