geometric_kernels.jax
=====================

.. py:module:: geometric_kernels.jax

.. autoapi-nested-parse::

   Loads JAX backend in lab, spherical_harmonics and geometric_kernels.

   ..note::
       A tutorial on the JAX backend is available in the
       :doc:`backends/JAX_Graph.ipynb </examples/backends/JAX_Graph>` notebook.