jax.random.categorical
======================

.. currentmodule:: jax.random

.. autofunction:: categorical