Skip to content

Instantly share code, notes, and snippets.

View dionhaefner's full-sized avatar

Dion Häfner dionhaefner

View GitHub Profile
"""A 1-dimensional example of adaptive mesh refinement in JAX. In this case, a simple
implementation of quadrature.
Static shapes don't mean you can't do this. Heap allocation is *not* necessary!
Not extensively tested; any bugs leave a comment below.
"""
import functools as ft
from collections.abc import Callable