The type info currently there doesn't make much sense: https://github.com/arviz-devs/xarray-einstats/blob/main/src/xarray_einstats/numba.py#L13-L22