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().
jaxtify/xaj
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|