NNX migration prep (4/N): sharding tools and Linen<->NNX checkpoint utilities#3525

Closed

ecnal-cienet wants to merge 2 commits intomainfrom

feat/nnx-linen-converter-and-sharding-tools

Commits on Apr 20, 2026

NNX: add TrainState, model creation utilities, and training loop support
ecnal-cienet
committed
NNX: add sharding tools, Linen<->NNX converter, and checkpoint comparison utility
ecnal-cienet
committed