tud_lbm.config.jax_config
=========================

.. py:module:: tud_lbm.config.jax_config

.. autoapi-nested-parse::

   JAX configuration settings for TUD-LBM.

   This module provides centralized JAX configuration options.
   Import and call `configure_jax()` at the start of your script to apply settings.



Functions
---------

.. autoapisummary::

   tud_lbm.config.jax_config.configure_jax


Module Contents
---------------

.. py:function:: configure_jax(enable_x64: bool | None = None, disable_jit: bool | None = None) -> None

   Configure JAX settings.

   Call this function at the start of your script to apply JAX configuration.
   If parameters are None, the module-level defaults are used.

   :param enable_x64: Enable 64-bit precision. Defaults to module-level ENABLE_X64.
   :param disable_jit: Disable JIT compilation for debugging. Defaults to module-level DISABLE_JIT.