jax.nn.softmax
==============

.. currentmodule:: jax.nn

.. autofunction:: softmax