corridorkey-mrp-mlx/scripts
cmoyates 57c3aff913
feat(phase6): optimization, benchmarking, and tiled inference
- Cache nn.Upsample instances in DecoderHead/GreenFormer __init__
  (eliminated ~7 allocations per forward pass)
- Add mx.compile() support via load_model(compile=True)
- Benchmark harness: eager vs compiled, multi-resolution, parity checks
- Tiled inference with overlap blending for large images
- Profiling utilities with forced mx.eval for accurate timing
- Reference comparison script (scripts/compare_reference.py)
- 12 new tests (compiled consistency + tiling)
- README performance section with Apple Silicon guidance

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-01 06:24:10 -03:30
..
bench_mlx.py feat(phase6): optimization, benchmarking, and tiled inference 2026-03-01 06:24:10 -03:30
compare_reference.py feat(phase6): optimization, benchmarking, and tiled inference 2026-03-01 06:24:10 -03:30
convert_weights.py feat(phase3): PyTorch→MLX weight converter (#1) 2026-03-01 05:16:56 -03:30
dump_pytorch_reference.py feat(phase2): MLX decoder + refiner with parity tests 2026-03-01 05:00:00 -03:30
infer.py fix: resize inputs to model img_size, use infer_and_save in CLI 2026-03-01 06:11:57 -03:30