cuPyNumeric

cuPyNumeric is a Legate library that aims to provide a distributed and accelerated drop-in replacement for the NumPy API on top of the Legion runtime. Using cuPyNumeric you can do things like run the final example of the Python CFD course completely unmodified on 2048 A100 GPUs in a DGX SuperPOD and achieve good weak scaling.

cuPyNumeric works best for programs that have very large arrays of data that cannot fit in the memory of a single GPU or a single node and need to span multiple nodes and GPUs. While our implementation of the current NumPy API is still incomplete, programs that use unimplemented features will still work (assuming enough memory) by falling back to the canonical NumPy implementation.

Installation

cuPyNumeric is available from conda on the legate channel. See https://docs.nvidia.com/cupynumeric/latest/installation.html for details about different install configurations, or building cuPyNumeric from source.

Documentation

The cuPyNumeric documentation can be found here.

Contributing

See the discussion on contributing in CONTRIBUTING.md.

Contact

For technical questions about cuPyNumeric and Legate-based tools, please visit the community discussion forum.

If you have other questions, please contact us at legate(at)nvidia.com.

Note

This project, i.e., cuPyNumeric, is separate and independent of the CuPy project. CuPy is a registered trademark of Preferred Networks.