For consistency with `DatasetToChunks`, consider calling this `chunks`. You might also add a `split_vars` parameter? _Originally posted by @shoyer in https://github.com/google/xarray-beam/pull/31#r713514917_