argparse>=1.4.0
jax[cpu]==0.4.30
matplotlib==3.8.4
