xarray-contrib/xarray-tutorial

dask chunking tutorial outline

dcherian opened this issue · 3 comments

from the pangeo working meeting discussion with @mgrover1 @jmunroe @norlandrhagen

Here's an outline for an intermediate tutorial talking about dask chunking specifically for Xarray users


Motivation: why care about chunk size?

Keeping track

  • monitoring chunk sizes and num tasks throughout the pipeline using the repr
    • use some images
  • while output blocks may be small (say after a big reduction), intermediate blocks need not be.
  • So keep monitoring chunksizes (and tasks) throughout the pipeline.

Why is it important to choose appropriate chunks early in the pipeline?

  • Demonstrate that rechunking is not cheap in most cases

Specify chunks when reading data

  1. Avoid chunks="auto".
  2. Specifying chunks during data read
    • open_dataset
    • open_mfdataset
  3. Analysis vs storage chunks:
    • Dask chunks should be a multiple of chunks on disk
    • talk about aligning chunks with files stored on disk
    • @djhoese example

Agreed this would be great to document thoroughly. See also this relevant issue + discussion in rioxarray corteva/rioxarray#253

demonstrate relation between chunk size and computation time / number of tasks with a simple example?

  • maybe even memory usage

This would be huge! This comes up often in Satpy where users want to process satellite images on their local machine but they only have 8GB or 16GB of memory. If someone can make a good diagram showing chunks being processed by a worker thread/process and how changing the size of all chunks or number of workers contributes to the overall memory usage that would be such a help when explaining this to users.

forgot to cc @rybchuk!