Merge pull request #2411 from d2l-ai/master

Add syne_tune, flax, jaxlib to colab libs
d2l-ai · Dec 15, 2022 · 833972e · 833972e
2 parents 7047d10 + eecbef4
commit 833972e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/config.ini b/config.ini
@@ -191,8 +191,11 @@ replace_svg_url = img, http://d2l.ai/_images
 libs = mxnet, mxnet, -U mxnet-cu101==1.7.0
        mxnet, d2l, d2l==RELEASE
        pytorch, d2l, d2l==RELEASE
+       pytorch, syne_tune, syne-tune[gpsearchers]
        tensorflow, d2l, d2l==RELEASE
        jax, d2l, d2l==RELEASE
+       jax, flax, flax
+       jax, jaxlib, jaxlib
 
 
 [sagemaker]