jax==0.4.33
jaxlib==0.4.33
ml_dtypes==0.5.0
numpy==2.1.1
opt_einsum==3.4.0
scipy==1.14.1
