Closed
Description
Zygote complains with a scalar indexing is disallowed error
with #999c335.
My code does work without that change. Sorry, I don't have a MWE to share at the moment.
Maybe the problem could be that during the FW pass of my custom Lux model I move an array to the CPU before I use it to solve an ODE and then the solution back to the GPU?
In case it's impossible to tell what could cause the error, I can try to reduce my model to a MWE.
Feel free to close this issue if you think it's a duplicate of #126
Metadata
Metadata
Assignees
Labels
No labels