defines JAXArrayContext
Co-authored-by:
Alexandru Fikl <alexfikl@gmail.com>
Showing
- arraycontext/__init__.py 8 additions, 1 deletionarraycontext/__init__.py
- arraycontext/impl/jax/__init__.py 129 additions, 0 deletionsarraycontext/impl/jax/__init__.py
- arraycontext/impl/jax/fake_numpy.py 143 additions, 0 deletionsarraycontext/impl/jax/fake_numpy.py
- doc/array_context.rst 6 additions, 0 deletionsdoc/array_context.rst
- doc/conf.py 1 addition, 0 deletionsdoc/conf.py
Loading
Please register or sign in to comment