Ricky Chen comments

Repositories
Issues
Comments

Results 33 comments of


                                            Ricky Chen

Unable to use vmap on a function containing the ode solver

I likely won't have time to support this, but will gladly take a PR for it!

real is not implemented for tensors with non-complex dtypes

You should upgrade your PyTorch. The more recent versions (I think versions >= 1.6?) have .real implemented for non-complex tensor types. In the meantime, I'll work on a fix for...

Initial augmented state on time

Does the conversation from #166 help? The quantity you're seeing is for computing dL/dt0, which intuitively, has the opposite gradient direction compared to t1, because t0 *shortens* the integration time...