Kvikio

https://docs.rapids.ai/api/kvikio/stable/ 

https://github.com/rapidsai/kvikio 

import cupy as cp

import kvikio

import kvikio.zarr

a = cp.arange(100)

with kvikio.CuFile("test-file", "w") as f:

    f.write(a)

b = cp.empty_like(a)

with kvikio.CuFile("test-file", "r") as f:

    f.read(b)

path = "/tmp/zarr-cupy-nvcomp"

z = kvikio.zarr.open_cupy_array(store=path, mode="w", shape=(20,), chunks=(5,)) # similar to zarr.open_array()