geometric_kernels.jax

Loads JAX backend in lab, spherical_harmonics and geometric_kernels.

..note::

A tutorial on the JAX backend is available in the backends/JAX_Graph.ipynb notebook.