advertise jax support a bit more

Author: alexfikl

README.rst

@@ -17,7 +17,9 @@ implementations for:
 
 - numpy
 - `PyOpenCL <https://documen.tician.de/pyopencl/array.html>`__
+- `JAX <https://jax.readthedocs.io/en/latest/>`__
 - `Pytato <https://documen.tician.de/pytato>`__ (for lazy/deferred evaluation)
+  with backends for ``pyopencl`` and ``jax``.
 - Debugging
 - Profiling
 

doc/array_context.rst

@@ -8,6 +8,13 @@ The Array Context Abstraction
 Implementations of the Array Context Abstraction
 ================================================
 
+..
+    When adding a new array context here, make sure to also add it to and run
+    ```
+        doc/make_numpy_coverage_table.py
+    ```
+    to update the coverage table below!
+
 Array context based on :mod:`pyopencl.array`
 --------------------------------------------
 
@@ -20,8 +27,8 @@ Lazy/Deferred evaluation array context based on :mod:`pytato`
 .. automodule:: arraycontext.impl.pytato
 
 
-Array context :mod:`jax.numpy`
--------------------------------------------------------------
+Array context based on :mod:`jax.numpy`
+---------------------------------------
 
 .. automodule:: arraycontext.impl.jax
 

doc/index.rst

@@ -7,6 +7,7 @@ implementations for:
 
 - :mod:`numpy`
 - :mod:`pyopencl`
+- :mod:`jax.numpy`
 - :mod:`pytato` (for lazy/deferred evaluation)
 - Debugging
 - Profiling