Update README.md (#75)

Mention JAX bindings, correct outdated comment for torch
lab-cosmo · Oct 26, 2023 · 8b2bef7 · 8b2bef7
1 parent bf3439a
commit 8b2bef7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -33,7 +33,8 @@ Pre-built (https://pypi.org/project/sphericart/).
 
 ```bash
 pip install sphericart             # numpy version
-pip install sphericart[torch]      # including also the torch bindings (CPU-only)
+pip install sphericart[torch]      # including also the torch bindings
+pip install sphericart[jax]        # JAX bindings (CPU-only)
 ```
 
 Note that the pre-built packages are compiled for a generic CPU, and might be