Ensure pip versions of jaxlib/jax installed

lanl · Sep 21, 2022 · 4bf71e3 · 4bf71e3
1 parent 391f0e5
commit 4bf71e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/misc/conda/make_conda_env.sh b/misc/conda/make_conda_env.sh
@@ -46,7 +46,7 @@ EOF
 )
 # Requirements that cannot be installed via conda (i.e. have to use pip)
 NOCONDA=$(cat <<-EOF
-bm3d bm4d faculty-sphinx-theme py2jn colour_demosaicing ray[tune]
+flax bm3d bm4d faculty-sphinx-theme py2jn colour_demosaicing ray[tune]
 EOF
 )