Alejandro A. Stefan-Zavala

Line 33 `jax.config.update('jax_array', True)` in `windtunnel3d.py` throws an exception. Looks it should be removed. Executing the `windtunnel3d.py` example in a Google Colab instance throws this exception: ``` --------------------------------------------------------------------------- AttributeError Traceback...