Fix: Refactor LoRA checkpoint restoration and simplify NNX weight extraction #4050

RexBearIU wants to merge 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

jackyf/lora-ckpt-converterAI-Hypercomputer/maxtext:jackyf/lora-ckpt-converterCopy head branch name to clipboard