jax[cuda12]==0.9.2
numpy==2.4.6
matplotlib==3.11.1
PyYAML==6.0.3
scipy==1.17.1
