feat(nnx): support native Flax NNX PEFT/LoRA training loop - #4346

Closed

RexBearIU wants to merge 1 commit into

mainfrom

nnx-lora-support

Commits on Jul 16, 2026

feat(nnx): support native Flax NNX PEFT/LoRA training loop with pre-trained base weight restoration and optimized upfront conversion
RexBearIU
committed
7ee8331