Jax框架在不同版本间的随机数生成并不一致 —— jax.random

官方：
https://jax.readthedocs.io/en/latest/api_compatibility.html#numerics-and-randomness

具体的解释就是在不同的Jax版本之间，即使设置相同随机种子也不能保证生成的随机数一致，比如： `

jax.random.gumbel(jax.random.key(72))

在不同版本的Jax中表现可能不一致。