daniel-trejobanos

I now realise that your pip installation asks for a specific jax version, which is a bit problematic for me, given that I am running on a M1 and installed...

I managed to downgrade jax, but there is no jaxlib 0.1.60 available in condaforge, seems like it could be the source of this bug I get when trying to load...