xma.functional
==============

.. toctree::
   :maxdepth: 4

   xma.functional.bmm
   xma.functional.continuous_count
   xma.functional.cross_entropy
   xma.functional.fused_residual_add_rmsnorm
   xma.functional.gru
   xma.functional.rnn
   xma.functional.sequence_packing
   xma.functional.softmax
   xma.functional.swiglu

.. toctree::
   :maxdepth: 4

   xma.functional.fused_linear_cross_entropy
   xma.functional.rmsnorm
   xma.functional.swiglu_packed

.. automodule:: xma.functional
   :members:
   :show-inheritance:
   :undoc-members: