fhchl
fhchl t1_izjb0tf wrote
Reply to comment by martenlienen in [R] torchode: A Parallel ODE Solver for PyTorch by martenlienen
Aight! Thanks for the nice answers! I wish a good conference :)
fhchl t1_izif55h wrote
Reply to comment by martenlienen in [R] torchode: A Parallel ODE Solver for PyTorch by martenlienen
Is this feature of torchode of solving multiple ODEs at once over some batch dimension comparable to jax.vmapping over that dimension in diffrax?
fhchl t1_izi5ecp wrote
Nice work! Though diffrax is mentioned, it would be interesting to see a direct comparison between diffrax and torchode. Can you give some more details in how they differ in features and performance, apart from the library in which they are implemented?
fhchl t1_izjd2vy wrote
Reply to comment by martenlienen in [R] torchode: A Parallel ODE Solver for PyTorch by martenlienen
+1 for the paper on parallel in time! Are there any implementation of those algorithms for torchode or diffrax out there?