petitRADTRANS.retrieval.covariance

See Greco & Brandt 2016, Nasedkin et al 2023 and Rotman et al 2025 for more information on covariance matrices and fitting.

Rotman et al. 2025: 10.3847/1538-4357/adef04

Attributes

_GLOBAL_KERNEL_ALIASES
_LOCAL_KERNEL_ALIASES
_MODE_ALIASES

Functions

normalize_covariance_mode(→ str)
normalize_global_kernel_name(→ str)
normalize_local_kernel_name(→ str)
_as_wavelength_vector(→ jax.numpy.ndarray)
_pairwise_separations(→ jax.numpy.ndarray)
squared_exponential_kernel(→ jax.numpy.ndarray)
matern32_kernel(→ jax.numpy.ndarray)
localized_squared_exponential_kernel(→ jax.numpy.ndarray)
build_global_covariance_kernel(→ jax.numpy.ndarray)
build_local_covariance_kernel(→ jax.numpy.ndarray)

Module Contents

petitRADTRANS.retrieval.covariance._GLOBAL_KERNEL_ALIASES

petitRADTRANS.retrieval.covariance._LOCAL_KERNEL_ALIASES

petitRADTRANS.retrieval.covariance._MODE_ALIASES

petitRADTRANS.retrieval.covariance.normalize_covariance_mode(mode: str | None) → str

petitRADTRANS.retrieval.covariance.normalize_global_kernel_name(kernel_name: str | None) → str

petitRADTRANS.retrieval.covariance.normalize_local_kernel_name(kernel_name: str | None) → str

petitRADTRANS.retrieval.covariance._as_wavelength_vector(wavelengths: jax.typing.ArrayLike) → jax.numpy.ndarray

petitRADTRANS.retrieval.covariance._pairwise_separations(wavelengths: jax.typing.ArrayLike) → jax.numpy.ndarray

petitRADTRANS.retrieval.covariance.squared_exponential_kernel(wavelengths: jax.typing.ArrayLike, amplitude: jax.typing.ArrayLike, length_scale: jax.typing.ArrayLike) → jax.numpy.ndarray

petitRADTRANS.retrieval.covariance.matern32_kernel(wavelengths: jax.typing.ArrayLike, amplitude: jax.typing.ArrayLike, length_scale: jax.typing.ArrayLike) → jax.numpy.ndarray

petitRADTRANS.retrieval.covariance.localized_squared_exponential_kernel(wavelengths: jax.typing.ArrayLike, amplitude: jax.typing.ArrayLike, length_scale: jax.typing.ArrayLike, center: jax.typing.ArrayLike) → jax.numpy.ndarray

petitRADTRANS.retrieval.covariance.build_global_covariance_kernel(wavelengths: jax.typing.ArrayLike, amplitude: jax.typing.ArrayLike, length_scale: jax.typing.ArrayLike, kernel_name: str = 'squared_exponential') → jax.numpy.ndarray

petitRADTRANS.retrieval.covariance.build_local_covariance_kernel(wavelengths: jax.typing.ArrayLike, amplitude: jax.typing.ArrayLike, length_scale: jax.typing.ArrayLike, center: jax.typing.ArrayLike, kernel_name: str = 'squared_exponential') → jax.numpy.ndarray