warpSPHIntegratorsMethods, examples and practical guidance
Differentiable ODE integrators for PyTorch — 142 schemes across 11 families, from explicit RK through coupled fully-implicit collocation to exponential integrators.
142 schemes, 11 families
Start with decay and a damped oscillator, then compare accuracy, stability and measured work. Friendly method names appear alongside their exact enum values and coefficient references.
Solving stiff problems
Implicit methods solve stage equations. Jacobian-free Newton–Krylov (JFNK) uses Jacobian actions rather than a stored matrix; diagnostics report nonlinear convergence, RHS work and Krylov iterations.
Gradients and custom systems
Begin with a tensor ODE or use the state protocol for coupled fields and stage-dependent quantities. Each family documents its gradient, operator and history assumptions, with runnable examples.