geometric_kernels.torch ======================= .. py:module:: geometric_kernels.torch .. autoapi-nested-parse:: Loads PyTorch backend in lab, spherical_harmonics and geometric_kernels. ..note:: A tutorial on the JAX backend is available in the :doc:`backends/PyTorch_Graph.ipynb ` notebook.