install jax for cpu

Author: mmcky

lectures/jax_intro.md

@@ -13,6 +13,13 @@ kernelspec:
 
 # An Introduction to JAX
 
+In addition to what's in Anaconda, this lecture will need the following libraries:
+
+```{code-cell} ipython3
+:tags: [hide-output]
+
+!pip install jax
+```
 
 This lecture provides a short introduction to [Google JAX](https://github.com/google/jax).