grid_budgeting

Functions

build_grid_budget_entries	Expand a grid-budget config into concrete MIP constraint entries.
run_grid_budgeted_mip

build_grid_budget_entries(grid_cfg)

Expand a grid-budget config into concrete MIP constraint entries.

Explicit entries are passed through, for example:

grid_budgeting:
  enabled: true
  entries:
    - name: depth_24
      mip_constraints: {"stats.not_no_op": 24}
    - name: params_8b
      human_constraints: {"num_params": "8B"}

Convenience fields map to the aggregate stats already computed by run_puzzle.py. Unknown future axes can be expressed through entries without changing this function.

Parameters:

grid_cfg (dict[str, Any])

Return type:

list[dict[str, Any]]

run_grid_budgeted_mip(hydra_cfg, grid_cfg)

Parameters:

• hydra_cfg (Any)

• grid_cfg (dict[str, Any])

Return type:

list[str]