configure_runtime

jeanspy.model_jax.configure_runtime

jeanspy.model_jax.configure_runtime(*, jax_enable_x64=None)

Update global runtime knobs used by model_jax.

Only JAX precision remains process-wide. Numerical integration choices are passed per call to DSphModel.sigmalos2 or ConstantAnisotropyModel.kernel.

Notes

Inputs and units. configure_runtime accepts jax_enable_x64; get_runtime_config takes no arguments. Device selection should be set through environment variables before JAX initialization.

Returns and shape. Dictionary of effective runtime settings and numerical defaults.

Validity. Changing precision after tracing can create a different numerical analysis and restart identity.

Errors. Unsupported keyword arguments raise TypeError.

Backend. Host configuration of JAX.

Differentiation. Configuration choices are static, not differentiable.

Examples. examples/docs_jax_spherical.py

Parameters:

jax_enable_x64 (bool | None)

Return type:

Dict[str, Any]