corridorkey-mrp-mlx/prompts/phase3-conversion.md
cmoyates 897242d200
init: repo scaffolding, deps, prompts, plan
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-01 04:48:51 -03:30

841 B

Work only on checkpoint conversion.

Goal: Create a robust conversion pipeline from the PyTorch CorridorKey checkpoint to MLX-compatible weights.

Requirements:

  • Inspect state_dict keys and map them explicitly.
  • Convert conv weights from PyTorch layout to MLX layout.
  • Preserve the patched 4-channel first conv behavior exactly.
  • Write converter diagnostics:
    • source key
    • destination key
    • source shape
    • destination shape
    • transform applied
  • Save output as safetensors or npz.
  • Validate load with MLX strict loading wherever possible.

Do not:

  • attempt full end-to-end model parity yet if Hiera is incomplete
  • hide key mismatches
  • use silent fallbacks

Definition of done:

  • converter script exists
  • mapping file exists
  • shape validation passes for completed modules
  • conversion report is readable and auditable