GPU arrays? Deferred-evaluation arrays? Just plain numpy arrays? You'd like your code to work with all of them? No problem! Comes with pre-made array context implementations for:
- numpy
- PyOpenCL
- JAX
- Pytato (for lazy/deferred evaluation) with backends for
pyopenclandjax. - Debugging
- Profiling
arraycontext started life as an array abstraction for use with the meshmode unstrucuted discretization package.
Distributed under the MIT license.