google-research / torchsde

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

Dev methods fixes #73

Closed patrick-kidger closed 3 years ago

patrick-kidger commented 3 years ago

Ticking off a variety of smaller to-dos from #15 :

The tests in test_sdeint.py are very numerous (~2000) covering all the common combinations of inputs, but are quick to run (~3 minutes); I felt it was best to be comprehensive on this central part of the library.

Thanks for your email btw - fortunately we've been doing different things, but I should have let you know!

lxuechen commented 3 years ago

I think we should also be able to merge problems.py and basic_sde.py now. I'd suggest keeping Ex1-4 and BasicSDE1-4. No pressure in doing it in this PR. I can also send in another PR on this issue.

patrick-kidger commented 3 years ago

I agree merging them makes sense. I've left them alone for now though.