xarray# Convert labelled xarray data to a Dataset and map predictions back onto the grid. Requires the optional extra: pip install "gpjax[xarray]". from_xarray Flatten labelled xarray data into a Dataset. GridSpec The labelled grid behind a flattened Dataset.