-
-
Notifications
You must be signed in to change notification settings - Fork 31
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
using parallel jax solver #61
Comments
Fantastic thanks. Will you be joining the dev meeting later? I will play with it now and may have a few questions |
I've got a shortlisting meeting that should end at 2pm, I can join after that or give you a call once its finished |
@martinjrobins Can Jax perform in parallel using multiple machines? |
I tried to run the Jax example that @martinjrobins posted above but I get the following error:
|
@wigging yes we changed how timescale is implemented recently in pybamm. Can you run it with the suggested option for a scalar timescale |
Using |
@TomTranter : here is an example script that runs 20 instances of the spm model in parallel using jax. Let me know if this is what you need.
The text was updated successfully, but these errors were encountered: