Max Lindqvist

Max Lindqvist

I model the physics of plasmas

cunumpy

Install from pypi with

pip install cunumpy

Simple wrapper for numpy and cupy. Replace import numpy as np with import cunumpy as xp.

Example usage:

export ARRAY_BACKEND=cupy

import cunumpy as xp
arr = xp.array([1,2])

print(type(arr))
print(xp.__version__)

Check out the code on Github!

Links:

  • https://pypi.org/project/cunumpy/