search

DiffEqML / torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
https://torchdyn.org
Apache License 2.0
1.33k stars 124 forks source link

Parallel integration routines with different t_spans ? #196

Open YouvenZ opened 11 months ago

YouvenZ commented 11 months ago

Thanks for this really amazing package!

I would like to know if the feature associated with parallel integration routines with different t_spans for each sample in a batch is being considered to be added in the package? I tried to find on the experimental branch as mentioned by the comment in the odeint.py script, however I did not find it.