google-research / torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
Apache License 2.0
1.56k stars 196 forks source link

Bugfixes for log-ODE #45

Closed patrick-kidger closed 4 years ago

patrick-kidger commented 4 years ago

Note that I'm looking to merge this into dev-log-ode.

*The RecursionErrors here are a good indication that trampolining / tail recursion is probably worth adding to BInterval. I'll probably focus on other things(=running experiments for the paper) in the short term, but I'll aim to get that added in time for our next release.

lxuechen commented 4 years ago

Thanks for the PR! Though, I actually have this access order fixed in my PR locally (not pushed). Would it make sense for me to push this fix in #43 after merging #44?

patrick-kidger commented 4 years ago

Whatever you like really!