Favorite way to go from netCDF (&xarray) to torch/TF/Jax et al

Oo, shiny! Yes I’ve got a GPU, let me test that out :smiley:

Edit: I’ve documented the installation/setup commands to try out @dcherian’s xr.open_dataset(store, engine="kvikio") proof of concept here if anyone is interested. Very hacky stuff but it works!

Edit 2: New blog post out on going from Zarr stores to GPU-backed xarray objects! Read it at Enabling GPU-native analytics with Xarray and kvikIO :partying_face:

4 Likes