REVERT: install jax from Pip

mmcky · mmcky · commit 08a15c8f74bd · 2022-03-17T19:40:19.000+11:00
diff --git a/lectures/back_prop.md b/lectures/back_prop.md
@@ -16,7 +16,7 @@ kernelspec:
 ```{code-cell} ipython3
 :tags: [hide-output]
 
-!conda install -y -c conda-forge jax=0.3.2 jaxlib
+!pip install --upgrade jax jaxlib
 !conda install -y -c plotly plotly plotly-orca retrying
 ```
 
diff --git a/lectures/opt_transport.md b/lectures/opt_transport.md
@@ -44,7 +44,7 @@ several implementations of linear programming, including, in order,
 ```{code-cell} ipython3
 :tags: [hide-output]
 !pip install --upgrade quantecon
-!pip install --upgrade POT
+!pip install --upgrade POT jax jaxlib
 ```
 
 Let's start with some imports.
