CuPy-Xarray: Xarray on GPUs!

GitHub Workflow CI Status pre-commit.ci status Documentation Status license

PyPI Conda-forge

NASA-80NSSC22K0345

Overview

CuPy-Xarray is a Python library that leverages CuPy, a GPU array library, and Xarray, a library for multi-dimensional labeled array computations, to enable fast and efficient data processing on GPUs. By combining the capabilities of CuPy and Xarray, CuPy-Xarray provides a convenient interface for performing accelerated computations and analysis on large multidimensional datasets.

Installation

CuPy-Xarray can be installed using pip or conda:

From Conda Forge:

conda install cupy-xarray -c conda-forge

From PyPI:

pip install cupy-xarray

The latest version from Github:

pip install git+https://github.com/xarray-contrib/cupy-xarray.git

Acknowledgements

Large parts of this documentations comes from SciPy 2023 Xarray on GPUs tutorial and this NCAR tutorial to GPUs.

Contents

User Guide:

Tutorials & Presentations:

Tutorials & Presentations

Contributing:

Contributing

Reference: