argparse>=1.4.0
jax[cpu]<0.6.1,>=0.4.30
matplotlib>=3.8.4
