geometric_kernels.torch

Loads PyTorch backend in lab, spherical_harmonics and geometric_kernels.

..note::

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