numpy==2.4.4 scipy==1.17.1 sympy==1.14.0 matplotlib==3.10.9 scikit-learn==1.9.0 mdtraj==1.11.1.post2 jax==0.10.2 jaxlib==0.10.2