petitRADTRANS.jax_dtype#
Single source of truth for petitRADTRANS JAX floating-point dtypes.
Import WORKING_JAX_DTYPE for memory-sensitive storage and routine working
arrays. Import FEAUTRIER_JAX_DTYPE for the Feautrier scattering solver and
other numerically sensitive kernels that should honor jax_enable_x64.
JAX_DTYPE is kept as a backward-compatible alias for
FEAUTRIER_JAX_DTYPE while the codebase is migrated to explicit per-surface
dtype choices.
Attributes#
Module Contents#
- petitRADTRANS.jax_dtype.WORKING_JAX_DTYPE#
- petitRADTRANS.jax_dtype.FEAUTRIER_JAX_DTYPE#
- petitRADTRANS.jax_dtype.FEAUTRIER_JAX_DTYPE#
- petitRADTRANS.jax_dtype.JAX_DTYPE#