numpy
scipy
jaxlib>=0.4.1
jax>=0.4.1
jaxopt>=0.5.5backends>=1.4.32
mlkernels>=0.3.6
