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.