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().
-
Notifications
You must be signed in to change notification settings - Fork 1
Numerical integrators for JAX
License
adxsrc/xaj
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Numerical integrators for JAX
Resources
License
Stars
Watchers
Forks
Packages 0
No packages published