maxtext.experimental.agent.ckpt_conversion_agent.step2 module# main script to execute the multi-agent workflow for model-specific mappings generation