Include JAX installation instructions

BYUCamachoLab · Jan 6, 2024 · e8e9539 · e8e9539
1 parent c63c9d8
commit e8e9539
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/docs/index.md b/docs/index.md
@@ -26,6 +26,11 @@ SiPANN provides additional models for use in your circuits. View the links for
 installation instructions and find out more. 
 :::
 
+If you are using a machine with a GPU, you should follow the instructions for
+[JAX installation](https://jax.readthedocs.io/en/latest/installation.html) in 
+order to get the dependencies required to use the GPU version of JAX, instead 
+of just the CPU version, which is installed by default.
+
 **To get started using Simphony**, check out the
 [Introduction](tutorials/intro). Tutorials and API references are accessible
 through the sidebar navigation.