.. include:: ../../examples/array_backends/jax.rst