I'm not sure we have an example where we show how to work with the InferenceData data structure and/or xarray related data structures See this https://github.com/bambinos/bambi/issues/533#issuecomment-2323371255