Dishant Mewada
Dishant Mewada
Restarting runtime and running everything again worked for me.
The first cell automatically uninstalls NumPy 1.25.2 and installs NumPy 1.26.4, so you don't need to install numpy again. When I run first two cells, it does throw the error...
Getting the same error. I used to restart the kernel after installing dependencies on colab and it used to work. But now after running the first cell and restarting the...
@Chi-chicken Can you please clarify how to do that? Thank you. I checked my jax.random file on local machine. It doesn't contain PRNGKeyArray. Only PRNGKey. Attached is my jax.random python...
@Chi-chicken Thank you for the reply. Can you please tell where is the file and what did you change in it?
You are such a saviour @Chi-chicken. Thank you so much. It is working on my local machine. I know the original question from @kaninaba94 was related to Google Colab. But...
I solved it by changing PRNGKeyArray to PRNGKEY in the initializers.py file. #1043 You can see the steps there to know where the file is located and what to change.