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
I keep experiencing relatively long compile times. I suspect, the Python loops in LieselInterface.update_state, and possibly in LieselInterface.extract_position might be partly responsible for this, see this point in the Jax FAQ. I have not looked into it in detail, but I hope these loops can be replaced by using something like jax.lax.scan.
The text was updated successfully, but these errors were encountered:
I keep experiencing relatively long compile times. I suspect, the Python loops in
LieselInterface.update_state
, and possibly inLieselInterface.extract_position
might be partly responsible for this, see this point in the Jax FAQ. I have not looked into it in detail, but I hope these loops can be replaced by using something likejax.lax.scan
.The text was updated successfully, but these errors were encountered: