RecodeX 重构消息,Flax模型序列化可使用nnx.to_flat_state将嵌套状态扁平化,再通过Safetensors的flax API存储为点分隔键的简单字典格式。