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.35k stars 125 forks source link

adding the ability to have parameters for each `step` #142

Closed joglekara closed 2 years ago

joglekara commented 2 years ago

e.g. a forcing function