Skip to content

Instantly share code, notes, and snippets.

@RicardoDominguez
RicardoDominguez / jax_bvp_solver.py
Last active June 11, 2024 17:41
BVP solver in JAX based on scipy.integrate.solve_bvp
"""Boundary value problem solver."""
import jax
import jax.numpy as jnp
# ------------------------------------------------------------------------------------------
# Linear solver for bordered almost block diagonal (BABD) systems
# ------------------------------------------------------------------------------------------
# Implementation as described in [1] Section 2.1 (structural orthogonal factorization).