traveler.skims.omx.OmxSkims.to_jax_arrays_3d

OmxSkims.to_jax_arrays_3d()

Load selected OMX matrices into grouped array objects.

Returns:

Arrays keyed by matrix base name. The concrete array type follows raw_array_type supplied at initialization, and time-dependent matrices use the configured time periods as their third dimension.

Return type:

dict[str, jax.Array or numpy.ndarray]

Raises:

ValueError – If shared memory is too small, a time-period suffix is unknown, or an OMX matrix has an incompatible shape.