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
pyopenclandjax. - Debugging
- Profiling
arraycontext started life as an array abstraction for use with the
meshmode unstrucuted discretization
package.
Distributed under the MIT license.