maxtext.models package
======================

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

Submodules
----------

.. toctree::
   :maxdepth: 4

   maxtext.models.deepseek
   maxtext.models.deepseek_batchsplit
   maxtext.models.deepseek_batchsplit_fp8
   maxtext.models.gemma
   maxtext.models.gemma2
   maxtext.models.gemma3
   maxtext.models.gemma4
   maxtext.models.gemma4_vision
   maxtext.models.gpt3
   maxtext.models.gpt_oss
   maxtext.models.llama2
   maxtext.models.llama4
   maxtext.models.mistral
   maxtext.models.mixtral
   maxtext.models.models
   maxtext.models.olmo3
   maxtext.models.qwen2
   maxtext.models.qwen3
   maxtext.models.qwen3_5
   maxtext.models.qwen3_custom
   maxtext.models.simple_layer