maxtext.layers.multi\_token\_prediction module
==============================================

.. automodule:: maxtext.layers.multi_token_prediction
   :members:
   :show-inheritance:
   :undoc-members: