NNX: add checkpoint comparison utility for Linen vs NNX validation

  Add a tool to compare checkpoint tree structures, shapes, and values
  across Linen and NNX formats.Supports cross-format and same-format comparisons with auto-detection, layer axis transposition, and RNG filtering.

Author: ecnal-cienet