apply_runtime_env

exo_skryer.help_runtime.apply_runtime_env(rc)

Apply optional rc.runtime settings before importing JAX / NumPyro.

Expected (all optional):

runtime:

platform: cpu | gpu | cuda | metal cuda_visible_devices: “0” or “0,1” preallocate: true/false mem_fraction: 0.8 jax_threads: 4 blackjax_threads: 4 numpyro_threads: 4 numpyro_platform: cpu | gpu # optional override for NumPyro/JAX device jax_x64: true/false

Returns:

numpyro_threadsint | None

Hint for numpyro.set_host_device_count in driver implementation.