maxtext.experimental.agent.ckpt_conversion_agent.dsl module

maxtext.experimental.agent.ckpt_conversion_agent.dsl module#

A DSL agent class, to propose potential new DSL rules/ops based on the previous analysis call.

class maxtext.experimental.agent.ckpt_conversion_agent.dsl.DSLAgent(api_key, dir_path, target_model='gemma3-4b', max_retries=3)[source]#

Bases: BaseAgent

DSL Agent

verify_dsl()[source]#

Verify DSL.