maxtext.layers package
======================

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

Submodules
----------

.. toctree::
   :maxdepth: 4

   maxtext.layers.attention_mla
   maxtext.layers.attention_op
   maxtext.layers.attentions
   maxtext.layers.decoders
   maxtext.layers.embeddings
   maxtext.layers.encoders
   maxtext.layers.engram
   maxtext.layers.initializers
   maxtext.layers.learn_to_init_layer
   maxtext.layers.linears
   maxtext.layers.mhc
   maxtext.layers.moe
   maxtext.layers.multi_token_prediction
   maxtext.layers.nnx_decoders
   maxtext.layers.nnx_wrappers
   maxtext.layers.normalizations
   maxtext.layers.pipeline
   maxtext.layers.pipeline_deprecated
   maxtext.layers.quantizations
   maxtext.layers.train_state_nnx