Ricky Chen

It seems the issue was updated. Right now, the error is func outputs a tuple but the initial state is a tensor. Their shapes need to match. (Error message could...

Our implementation is written purely in PyTorch, so it just conforms to however PyTorch is executed. That is, if you run it normally it only allocates one process. But if...

Yeah, just define it anywhere. In order to use `odeint_adjoint`, it's good practice to define them as part of the module. ``` global_params = ... class ODEfunc(nn.Module): def __init__(self): self.parameters...

I've just enabled complex types on the latest commit. Give it a try and if there are issues, let me know. To install the latest commit: ``` pip install git+https://github.com/rtqichen/torchdiffeq...

The current API isn't designed with this mind unfortunately. The only way to implement it is to use the internal functions. The equivalent to the `step` method is the `_advance`...

No sorry, zero progress has been made since 2019. If anyone wants to submit a PR for this, I can approve it.