absl-py==2.2.2
aiohappyeyeballs==2.6.1
aiohttp==3.11.18
aiosignal==1.3.2
annotated-types==0.7.0
antlr4-python3-runtime==4.9.3
array-record==0.7.1
ase==3.24.0
asttokens==3.0.0
astunparse==1.6.3
attrs==25.3.0
cachetools==5.5.2
certifi==2025.1.31
charset-normalizer==3.4.1
cheroot==10.0.1
chex==0.1.89
click==8.1.8
cloudpickle==3.1.1
comm==0.2.2
contourpy==1.3.1
cycler==0.12.1
dataclasses==0.8
debugpy==1.8.13
decorator==5.2.1
dm-haiku==0.0.13
dm-tree==0.1.9
docker-pycreds==0.4.0
docstring-parser==0.16
e3nn-jax==0.20.3
einops==0.8.1
equinox==0.11.10
etils==1.12.2
executing==2.2.0
filelock==3.18.0
flatbuffers==25.2.10
flax==0.10.4
fonttools==4.57.0
frozendict==2.4.6
frozenlist==1.6.0
fsspec==2025.3.2
gast==0.6.0
gcsfs==2025.3.2
gitdb==4.0.12
gitpython==3.1.44
google-api-core==2.24.2
google-auth==2.39.0
google-auth-oauthlib==1.2.2
google-cloud-core==2.4.3
google-cloud-storage==3.1.0
google-crc32c==1.7.1
google-pasta==0.2.0
google-resumable-media==2.7.2
googleapis-common-protos==1.70.0
grpcio==1.72.0
gviz-api==1.10.0
h5py==3.13.0
humanize==4.12.2
idna==3.10
immutabledict==4.2.1
importlib-resources==6.5.2
iniconfig==2.1.0
ipykernel==6.29.5
ipython==8.34.0
jaraco-functools==4.1.0
jax==0.4.29
jax-cuda12-pjrt==0.4.29
jax-cuda12-plugin==0.4.29
jax-md @ git+https://github.com/jax-md/jax-md.git@3571f47bd37a4a1d316d68797ed16f52add5e743
jax-sph==0.1.0
jaxlib==0.4.29
jaxopt==0.8.3
jaxtyping==0.3.1
jedi==0.19.2
jinja2==3.1.6
jmp==0.0.4
joblib==1.4.2
jraph==0.0.6.dev0
jupyter-client==8.6.3
jupyter-core==5.7.2
keras==3.9.2
kiwisolver==1.4.8
latexcodec==3.0.0
libclang==18.1.1
lineax==0.0.7
markdown==3.8
markdown-it-py==3.0.0
markupsafe==3.0.2
matplotlib==3.10.1
matplotlib-inline==0.1.7
matscipy==1.1.1
mdurl==0.1.2
ml-collections==1.0.0
ml-dtypes==0.5.1
monty==2025.3.3
more-itertools==10.7.0
mpmath==1.3.0
msgpack==1.1.0
multidict==6.4.3
namex==0.0.9
narwhals==1.34.0
nest-asyncio==1.6.0
networkx==3.4.2
numpy==1.26.4
nvidia-cublas-cu12==12.4.5.8
nvidia-cuda-cupti-cu12==12.4.127
nvidia-cuda-nvcc-cu12==12.8.93
nvidia-cuda-nvrtc-cu12==12.4.127
nvidia-cuda-runtime-cu12==12.4.127
nvidia-cudnn-cu12==9.1.0.70
nvidia-cufft-cu12==11.2.1.3
nvidia-curand-cu12==10.3.5.147
nvidia-cusolver-cu12==11.6.1.9
nvidia-cusparse-cu12==12.3.1.170
nvidia-cusparselt-cu12==0.6.2
nvidia-ml-py==12.570.86
nvidia-nccl-cu12==2.21.5
nvidia-nvjitlink-cu12==12.4.127
nvidia-nvtx-cu12==12.4.127
oauthlib==3.2.2
omegaconf==2.3.0
opt-einsum==3.4.0
optax==0.2.4
optree==0.15.0
orbax-checkpoint==0.6.4
ott-jax==0.4.6
packaging==24.2
palettable==3.3.3
pandas==2.2.3
parso==0.8.4
pexpect==4.9.0
pillow==11.1.0
platformdirs==4.3.7
plotly==6.0.1
pluggy==1.5.0
pooch==1.8.2
promise==2.3
prompt-toolkit==3.0.50
propcache==0.3.1
proto-plus==1.26.1
protobuf==4.25.7
psutil==7.0.0
ptyprocess==0.7.0
pure-eval==0.2.3
pyarrow==19.0.1
pyasn1==0.6.1
pyasn1-modules==0.4.2
pybtex==0.24.0
pydantic==2.11.2
pydantic-core==2.33.1
pygments==2.19.1
pymatgen==2025.3.10
pyparsing==3.2.3
pytest==8.3.5
python-dateutil==2.9.0.post0
pytz==2025.2
pyvista==0.44.2
pyyaml==6.0.2
pyzmq==26.4.0
requests==2.32.3
requests-oauthlib==2.0.0
rich==14.0.0
rsa==4.9.1
ruamel-yaml==0.18.10
ruamel-yaml-clib==0.2.12
scipy==1.15.2
scooby==0.10.0
sentry-sdk==2.25.1
setproctitle==1.3.5
setuptools==78.1.0
simple-parsing==0.1.7
six==1.17.0
smmap==5.0.2
spglib==2.6.0
stack-data==0.6.3
sympy==1.13.1
tabulate==0.9.0
tensorboard==2.19.0
tensorboard-data-server==0.7.2
tensorboard-plugin-profile==2.19.4
tensorflow==2.19.0
tensorflow-datasets==4.9.8
tensorflow-io-gcs-filesystem==0.37.1
tensorflow-metadata==1.17.1
tensorstore==0.1.73
termcolor==3.0.1
toml==0.10.2
toolz==1.0.0
torch==2.6.0
tornado==6.4.2
tqdm==4.67.1
traitlets==5.14.3
treescope==0.1.9
triton==3.2.0
typing-extensions==4.13.1
typing-inspection==0.4.0
tzdata==2025.2
uncertainties==3.2.2
urllib3==2.3.0
vtk==9.3.1
wadler-lindig==0.1.4
wandb==0.19.9
wcwidth==0.2.13
werkzeug==3.1.3
wget==3.2
wheel==0.45.1
wrapt==1.17.2
yarl==1.20.0
zipp==3.21.0
