Håkon
Håkon
I am running in Google Colab, and am still seeing the issue module 'jax' has no attribute 'xla' I have been running the exact same script for a couple of...
Sure. It is not exactly optimized, with some commented out sections, but it is essentially a version of one of your templates, made to connect with google drive and get...
Thank you very much, it seems to be working now.
Sorry, I was being stupid. That makes sense, if in addition you have special handlers for the beginning and the end, since I see the values for the different A's...
Thank you, that explains it.