search

rtqichen / torchdiffeq

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

Export to ONNX? #235

Open bryanhpchiang opened 1 year ago

bryanhpchiang commented 1 year ago

Would it be possible to export a nn.Module that uses odeint under the hood to ONNX?