Coverage for geometric_kernels/jax.py: 100%

1""" 

2Loads JAX backend in lab, spherical_harmonics and geometric_kernels. 

3 

4..note:: 

5 A tutorial on the JAX backend is available in the 

6 :doc:`backends/JAX_Graph.ipynb </examples/backends/JAX_Graph>` notebook. 

7""" 

8 

9import logging 

10 

11import lab.jax # noqa 

12import spherical_harmonics.jax # noqa 

13 

14import geometric_kernels.lab_extras.jax # noqa 

15 

16logging.getLogger(__name__).info("JAX backend enabled.")