RecodeX 重构消息,作者通过哈希值调试发现JAX/Flax NNX训练循环中参数未被更新,原因是错误使用@jax.jit而非@nnx.jit导致梯度无法应用,修正后损失正常下降。