Open martinjrobins opened 3 years ago
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:
ValueError: model.timescale must be a Scalar after parameter processing
(cannot contain 'InputParameter's). You have probably set one of the
parameters used to calculate the timescale to an InputParameter. To avoid
this error, hardcode model.timescale to a constant value by passing the
option {'timescale': value} to the model.
@wigging yes we changed how timescale is implemented recently in pybamm. Can you run it with the suggested option for a scalar timescale
Using model = pybamm.lithium_ion.SPM(options={"timescale": 1.0})
makes the example work.
@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.