float64

anri.crystal.float64()

Evaluate JAX in float64 inside a with block, without changing the global setting.

Returns:

AbstractContextManager – jax.enable_x64(True), or jax.experimental.enable_x64() on older JAX