yippy._precision

Precision policy for yippy.

All float storage follows the global jax_enable_x64 flag: float32 when the flag is off (the default, memory-friendly), float64 when it is on. This avoids a segmented float32/float64 pipeline, which JAX handles poorly.

Functions

float_dtype()	Active default float dtype: float64 if jax_enable_x64 else float32.
dtype_tag()	Short cache key for the active float dtype: "f32" or "f64".

Module Contents

yippy._precision.float_dtype()

Active default float dtype: float64 if jax_enable_x64 else float32.

Returns a numpy dtype usable for both np.* and jnp.* allocations.

yippy._precision.dtype_tag()

Short cache key for the active float dtype: "f32" or "f64".