search

rtqichen / torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
MIT License
5.61k stars 930 forks source link

`event_fun` with multiple initial conditions #221

Closed roflmaostc closed 1 year ago

roflmaostc commented 1 year ago

Hi!

I was following this for the event_fun.

Is it possible to simulate N bouncing balls with different initial conditions? https://github.com/rtqichen/torchdiffeq/blob/master/examples/bouncing_ball.py

Hence, the stopping time could be different for different balls.

I didn't find an example and would be happy about any tipps :)!

Best,

Felix

roflmaostc commented 1 year ago

Closed because of dup #201