Make sure package is built with dask compatibility
Closed this issue · 5 comments
tdmorello commented
Make sure package is built with dask compatibility
yfukai commented
I think this will be useful for lazy-loading dataset we discussed. With dask compatibility, we might even not need our custom image loading functionality.
yfukai commented
jax-ml/jax#395
It seems that JAX needs the data to be on-memory for now.
yfukai commented
For transform, we can definitely make it compatible with dask.