MoMo

> JAX arrays on CPU use FTZ mode, that is, subnormal numbers (like 1e-45) are flushed to zeros: > > ```python > >>> with jax.default_device(jax.devices('cpu')[0]): > ... jax.numpy.array(numpy.array([-0.0, 1.401298464324817e-45, -0.0],...

> In processors that support the FTZ flag, enabling FTZ is optional. PyTorch obviously does not enable FTZ mode while JAX (read: some of its underlying component) does on CPU....

> We confirmed in related issue (#24281) that JAX handles signed zeros consistently in operations like `argsort` and `argmax`. > > With that out of the way, it seems your...

> [我怀疑这与#24275](https://github.com/jax-ml/jax/issues/24275)中看到的是同一件事：JAX 以 32 位计算，而您正在与以 64 位计算的平台进行比较。 But please look at my latest code, I use float32, but the final result of jax is obviously different from the results of...