CuPy-Xarray: Xarray on GPUs!

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:

User Guide

• Basics of CuPy
• Introduction
• Basic Computations
• High-level Computation
• Custom Kernels with apply_ufunc
• A real world example

Tutorials & Presentations:

Tutorials & Presentations

• Tutorials and Presentations

Contributing:

Contributing

• Contributing guide

API Reference: