Skip to content

Instantly share code, notes, and snippets.

View avik-pal's full-sized avatar
😃
Focusing

Avik Pal avik-pal

😃
Focusing
View GitHub Profile
@RicardoDominguez
RicardoDominguez / jax_bvp_solver.py
Last active March 4, 2024 14:31
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).