colabtracemedia

Hi, thanks for the update. I now get this error in the Colab Inference Notebook when jax is imported: `import jax import jax.numpy as jnp` RuntimeError: jaxlib is version 0.3.25,...

Great, thank you, Inference Notebook now fixed!

Hi, it looks like there's now an error with the version of jaxlib when running the Inference Notebook:

Adding jaxlib and cuda version to setup.cfg install_requires works for Google Colab Inference notebook: `jaxlib@https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.3.25+cuda11.cudnn82-cp39-cp39-manylinux2014_x86_64.whl`