This came from a discussion in #8928, where [this message](https://github.com/dask/dask/issues/8928#issuecomment-1102606265) from @jsignell contains useful hints on how to start working on this.