unit\_scaling.functional.softmax ================================ .. currentmodule:: unit_scaling.functional .. autofunction:: softmax