numpy
jax
dsps
diffmah
dsps
flowjax