jaxopt
hydra-core
optax
pandas
