From 852666956d7b6bb3cfaebdfa158c6719743803a5 Mon Sep 17 00:00:00 2001 From: Filippo Bigi <98903385+frostedoyster@users.noreply.github.com> Date: Sun, 15 Feb 2026 17:44:24 +0100 Subject: [PATCH] !! Remove outdated info from docs !! Updated installation instructions to include JAX support and removed GPU support details. --- docs/src/installation.rst | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/docs/src/installation.rst b/docs/src/installation.rst index 66ea42efa..8b9e29ff6 100644 --- a/docs/src/installation.rst +++ b/docs/src/installation.rst @@ -11,16 +11,14 @@ The Python package can be installed with pip by simply running pip install sphericart -This basic package makes use of NumPy. A PyTorch-based implementation can be installed with +This basic package makes use of NumPy. Implementations supporting PyTorch and JAX can be installed with .. code-block:: bash pip install sphericart[torch] + pip install sphericart[jax] -This pre-built version available on PyPI sacrifices some performance to ensure it -can run on all systems, and it does not include GPU support. -If you need an extra 5-10% of performance, you want to evaluate the spherical harmonics on GPUs, -and/or you want to use it in JAX, you should build the code from source: +If you need an extra 5-10% of performance, you should build the code from source: .. code-block:: bash