Commit 4a574ea
committed
NNX: add bidirectional Linen<->NNX checkpoint conversion utility
Add a bidirectional Linen <-> NNX checkpoint converter tool that handles:
- Auto-detection of checkpoint format
- Conversion of params structure (double nesting vs flat)
- Stacking/unstacking per-layer parameters
- Value wrapper handling for NNX format1 parent bd6f280 commit 4a574ea
2 files changed
Lines changed: 1450 additions & 0 deletions
File tree
- src/maxtext/checkpoint_conversion
- tests/unit
0 commit comments