Skip to content

Custom Solvers Guide¶

Coming Soon

This guide is being written. Check back soon for Diffrax and DifferentialEquations.jl integration examples.

Overview¶

Use VectorBuilder to integrate custom ODE solvers: - JAX/Diffrax - Julia/DifferentialEquations.jl - Custom Python solvers

Basic Pattern¶

def custom_solver(params):
    # Your solver here
    # Return predictions at observation times
    return predictions

builder = (
    diffid.VectorBuilder()
    .with_objective(custom_solver)
    .with_data(data)
    .with_parameter("alpha", 1.0)
)

Examples¶

See the predator-prey examples: - predator_prey_diffrax.py - predator_prey_diffeqpy.py

See Also¶