# Tested Linux x86_64 / Python 3.12 / GPU dependency lock
# Install sources at commits in sources.json separately with --no-deps
absl-py==2.5.0
attrs==26.1.0
chex==0.1.92
cloudpickle==3.1.2
decorator==5.3.1
distrax==0.1.9
dm-tree==0.1.10
gast==0.7.0
h5py==3.16.0
immutabledict==4.3.1
iniconfig==2.3.0
jax==0.8.2
jax-cuda12-pjrt==0.8.2
jax-cuda12-plugin==0.8.2
jaxlib==0.8.2
jaxtyping==0.3.11
ml_collections==1.1.0
ml_dtypes==0.6.0
numpy==2.5.3
nvidia-cublas-cu12==12.9.2.10
nvidia-cuda-cccl-cu12==12.9.27
nvidia-cuda-cupti-cu12==12.9.79
nvidia-cuda-nvcc-cu12==12.9.86
nvidia-cuda-nvrtc-cu12==12.9.86
nvidia-cuda-runtime-cu12==12.9.79
nvidia-cudnn-cu12==9.26.0.51
nvidia-cufft-cu12==11.4.1.4
nvidia-cusolver-cu12==11.7.5.82
nvidia-cusparse-cu12==12.5.10.65
nvidia-nccl-cu12==2.32.3
nvidia-nvjitlink-cu12==12.9.86
nvidia-nvshmem-cu12==3.7.2
opt_einsum==3.4.0
optax==0.2.8
packaging==26.3
pandas==3.0.6
pluggy==1.6.0
pyblock==0.6
Pygments==2.21.0
pyscf==2.14.0
pytest==9.1.1
python-dateutil==2.9.0.post0
PyYAML==6.0.3
scipy==1.18.1
setuptools==83.0.0
six==1.17.0
tfp-nightly==0.26.0.dev20260922
toolz==1.1.0
typing_extensions==4.16.0
wadler_lindig==0.1.7
wheel==0.47.0
wrapt==2.4.1
