torchdiffeq
torchdiffeq copied to clipboard
Use Sherman-Morrison formula with implicit solvers
Mentioned in #265 , the Jacobian inverse is explicitly calculated for the FIRK and DIRK solvers. To avoid this expensive calculation, the Sherman-Morrison formula can be used to calculate and store the Jacobian inverse.