You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thanks for spotting this issue! You are right. We have a major refactoring coming in very soon (see branch major-refactoring) in which we have added a second backened to XLB based on Warp. We will fix this issue in that version. Please stay tuned @astefanz
Line 33
jax.config.update('jax_array', True)
inwindtunnel3d.py
throws an exception. Looks it should be removed.Executing the
windtunnel3d.py
example in a Google Colab instance throws this exception:This happens on an L4 instance. I install dependencies by executing, in Colab notebook cells:
And then cloning the repository and executing the example:
Looks like the operation in line 33 was removed after the array migration in JAX 0.4.1, per: https://jax.readthedocs.io/en/latest/jax_array_migration.html and this line should just be removed. Thanks!
The text was updated successfully, but these errors were encountered: