arraycontext: Choose your favorite numpy-workalike

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 pyopencl and jax.
• Debugging
• Profiling

arraycontext started life as an array abstraction for use with the meshmode unstrucuted discretization package.

Distributed under the MIT license.

Links

• Source code on Github
• Documentation