dependancy issues in the google colab

[KissakiBiosciences]

Hi,

I am getting all sorts of dependency issues in the google colab, probably between jax and numpy.

Do you have a stable environment I could try?

Thanks!