NNX: add checkpoint conversion utility and fix NNX checkpoint save/restore

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 format

Author: ecnal-cienet