colossalai.kernel.cuda\_native.scaled\_softmax
==============================================

.. automodule:: colossalai.kernel.cuda_native.scaled_softmax
   :members: