Merge pull request #2411 from d2l-ai/master

Add syne_tune, flax, jaxlib to colab libs

Author: astonzhang

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]