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
dtype errors with adaptive integrator #1480
Comments
Thanks, it all makes sense. In particular
I agree
I would initialise it with the dtype of the output of the Those two should be enough, I think? I don't think that there are other fields that might change... We already forcibly enforce that the dtype of the parameters does not change (for the same reason). |
Care to make a PR? |
When using an adaptive integrator and a dtype=jnp.float32 for parameters, samples, etc, the adaptive integrator raises errors. The errors arise due to the following:
I'm not sure what the best solution is (I wasn't even expecting errors of this kind tbh), but some possibilities are:
The text was updated successfully, but these errors were encountered: