XAJ: Numerical Integrators for JAX

XAJ is a high performance numerical integration framework for Google’s JAX. The new programming interface is designed to make integration composable with other JAX constructs such as just-in-time compilation (jit()) and vectorization (vmap()). Wrappers are available to make it compatible with scipy.integrate.solve_ivp().

Contents:

• Interface Design
  • Solution Callable
  • Composability
  • Call Signature
  • Dual Integrators
• Design Patterns
  • Stepping Engine
  • Dense Output
• Implementation
  • Stepper
  • Error Estimator
  • Stepsize Controller
  • Stepping Engine
  • Cache and Dense Output
  • Pacer
  • Treker
  • Solution

Indices and tables

• Index

• Module Index

• Search Page