maxtext.experimental.agent.ckpt_conversion_agent package

Subpackages

• 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.ground_truth package
  • Submodules
    • maxtext.experimental.agent.ckpt_conversion_agent.ground_truth.gemma3 module
      • GEMMA3_MAXTEXT_TO_HF_PARAM_HOOK_FN()
    • maxtext.experimental.agent.ckpt_conversion_agent.ground_truth.llama3 module
      • LLAMA31_MAXTEXT_TO_HF_PARAM_HOOK_FN()
    • maxtext.experimental.agent.ckpt_conversion_agent.ground_truth.qwen3 module
      • QWEN3_MAXTEXT_TO_HF_PARAM_HOOK_FN()
• maxtext.experimental.agent.ckpt_conversion_agent.utils package
  • Submodules
    • maxtext.experimental.agent.ckpt_conversion_agent.utils.save_param module
    • maxtext.experimental.agent.ckpt_conversion_agent.utils.utils module
      • load_prompt_template()
      • load_text_file()
      • load_json()
      • assemble_script()

Submodules

• maxtext.experimental.agent.ckpt_conversion_agent.analysis module
  • AnalysisAgent
    • AnalysisAgent.analyze_model_structures()
• maxtext.experimental.agent.ckpt_conversion_agent.base module
  • BaseAgent
    • BaseAgent.generate_text()
• maxtext.experimental.agent.ckpt_conversion_agent.dsl module
  • DSLAgent
    • DSLAgent.verify_dsl()
• maxtext.experimental.agent.ckpt_conversion_agent.evaluation module
  • compressed_size()
  • estimate_kolmogorov()
  • main()
• maxtext.experimental.agent.ckpt_conversion_agent.mapping module
  • MappingAgent
    • MappingAgent.generate_param_mapping()
    • MappingAgent.generate_shape_mapping()
• maxtext.experimental.agent.ckpt_conversion_agent.plan module
  • PlanAgent
    • PlanAgent.plan_conversion()
• maxtext.experimental.agent.ckpt_conversion_agent.prompt_chain module
  • prompt_chaining_agent
    • prompt_chaining_agent.run_chain()
• maxtext.experimental.agent.ckpt_conversion_agent.step1 module
• maxtext.experimental.agent.ckpt_conversion_agent.step2 module
• maxtext.experimental.agent.ckpt_conversion_agent.transformation module
  • TransformationAgent
    • TransformationAgent.generate_hook_functions()