jax
jaxopt
mhealpy
numba
