maxtext.experimental.agent.ckpt_conversion_agent.baselines package

Subpackages

• maxtext.experimental.agent.ckpt_conversion_agent.baselines.context package
  • Submodules
    • maxtext.experimental.agent.ckpt_conversion_agent.baselines.context.hf_shape module
      • GEMMA2_HF_WEIGHTS_TO_SHAPE()
    • maxtext.experimental.agent.ckpt_conversion_agent.baselines.context.param_mapping module
      • GEMMA2_MAXTEXT_TO_HF_PARAM_MAPPING()
      • GEMMA2_MAXTEXT_TO_HF_PARAM_HOOK_FN()