python -m math_jlens.explore \
--model ../Lightning-OPD/checkpoints/qwen3-4b-base-sft-qwen3-8b/ \
--data data/dapo-math-17k/dapo-math-17k.jsonl \
--lens outputs/multi-gpu-20/lens-bf16.pt \
--fit-count 20 \
--seed 17 \
--sample-index 0 \
--device cuda:4 | 
