colab demo error

[daviddaiiiii]

the demo colab pt1.introduction comes up with an error on the second running block
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
ipython 7.9.0 requires jedi>=0.10, which is not installed.

[daviddaiiiii]

also, the Velo colab demo shows the similar issue
RuntimeError: jaxlib is version 0.3.25, but this version of jax requires version >= 0.4.2.
guess the learned_optimaztion project jaxlib version needs to update